search

wenyu332 / daguan_bert_ner

30 stars 13 forks source link

训练bert出错，请教一下 #13

Open luluforever opened 5 years ago

luluforever commented 5 years ago

1

wenyu332 commented 5 years ago

你是多gpu训练是吧 损失改为loss.mean().item()就行

luluforever commented 5 years ago

是的，感谢老哥呀

wenyu332 commented 5 years ago

哈哈，好的 还有，这个多卡的我也没跑过，如果你训练后性能与单卡保持不变的话，后面在下面留下言吧。。

jieyang123 commented 5 years ago

您好，请问多gpu训练如何设置呀？

luluforever commented 5 years ago

您好，请问多gpu训练如何设置呀？

我发现我这个其实也是单卡在训练

luluforever commented 5 years ago

哈哈，好的 还有，这个多卡的我也没跑过，如果你训练后性能与单卡保持不变的话，后面在下面留下言吧。。

改了后，我这训练太慢了，一天就跑2个epoch,根本跑不完

wenyu332 commented 5 years ago

换新电脑吧 哈哈 @luluforever