search

liuyukid / transformers-ner

Pytorch-Named-Entity-Recognition-with-transformers
209 stars 44 forks source link

多GPU跑导致服务器重启 #8

Open 2hip3ng opened 4 years ago

2hip3ng commented 4 years ago

在单片gpu上跑没事，变成4卡就导致服务器重启，这是咋回事，这大概是咋回事。

liuyukid commented 4 years ago

这可真不是我代码的锅，你试试看双卡，小batch_size

2hip3ng commented 4 years ago

下次服务器没人用的时候再试试，你试过多卡跑吗？

liuyukid commented 4 years ago

试过啊，而且crf的代码我还专门调整过，可以在多gpu上跑