search

taishan1994 / BERT-BILSTM-CRF

使用BERT-BILSTM-CRF进行中文命名实体识别。
312 stars 38 forks source link

M2芯片使用mps训练速度极慢 #11

Open wongshg opened 10 months ago

wongshg commented 10 months ago

大佬，我把device的使用语句改成如图的mps，但是训练的速度极慢，感觉甚至比cpu慢十倍，GPU的占用也很低，但是内存占用挺多，9G左右。我测试了其他的mps使用场景是正常的，比cpu速度快好几倍。想请教下是否是我的设置问题。

Snipaste_2023-12-10_21-03-37 Snipaste_2023-12-10_21-13-45
taishan1994 commented 9 months ago

应该就只是这里改一下，没试过m2