search

taishan1994 / BERT-BILSTM-CRF

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

多卡训练 #4

Open moonyueqi opened 1 year ago

moonyueqi commented 1 year ago

请问大佬，如果要改成多卡训练，使用DataParallel方法该如何修改代码呢？

taishan1994 commented 1 year ago

请问大佬，如果要改成多卡训练，使用DataParallel方法该如何修改代码呢？

参考这里：https://github.com/taishan1994/pytorch-distributed-NLP