NER-LSTM-CRF icon indicating copy to clipboard operation
NER-LSTM-CRF copied to clipboard

请问可以改成多GPU的训练方式吗?

Open muzier opened this issue 7 years ago • 1 comments

感谢博主的无私分享! 目前想在多GPU平台上跑这个模型,但是看model.py里只有device:cpu这个变量,不知道如果改成多gpu的版本,还有哪些参数或设置需要更改啊? 先谢过啦!

muzier avatar Aug 27 '18 02:08 muzier

你好,这个还不支持多GPU训练,如果要改的话,你可以参考网上的教程,不是很复杂

On Mon, Aug 27, 2018 at 10:14 AM LiPeng [email protected] wrote:

感谢博主的无私分享!

目前想在多GPU平台上跑这个模型,但是看model.py里只有device:cpu这个变量,不知道如果改成多gpu的版本,还有哪些参数或设置需要更改啊? 先谢过啦!

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/liu-nlper/NER-LSTM-CRF/issues/27, or mute the thread https://github.com/notifications/unsubscribe-auth/ASWWiiv3iLiCkLg9caaDUdKn4hhEt_qzks5uU1YegaJpZM4WNEnC .

liu-nlper avatar Aug 27 '18 09:08 liu-nlper