search

axinc-ai / ailia-models-cpp

C++ version of ailia models repository
17 stars 2 forks source link

Implement bert maskedlm #21

Closed kyakuno closed 1 year ago

kyakuno commented 1 year ago

16

kyakuno commented 1 year ago 
kyakuno@KazukinoMacBook-Pro bert_maskedlm % ./bert_maskedlm
env_id : 0 type : 0 name : CPU
env_id : 1 type : 1 name : CPU-AppleAccelerate
env_id : 2 type : 2 name : MPSDNN-Apple M1 Max
you can select environment using -e option
selected env name : MPSDNN-Apple M1 Max
Input : 私は[MASK]で動く。
Input Tokens :
1325 9 4 12 11152 8 
Predictions :
0 単独 8.405500
1 高速 8.075030
2 自動 7.382397
3 屋内 7.289096
4 、 7.250325
Output : 私は単独で動く。
Output Tokens :
1325 9 3713 12 11152 8 
Program finished successfully.