920232796 / bert_seq2seq

pytorch实现 Bert 做seq2seq任务,使用unilm方案,现在也可以做自动摘要,文本分类,情感分析,NER,词性标注等任务,支持t5模型,支持GPT2进行文章续写。
Apache License 2.0
1.29k stars 209 forks source link

多卡训练失败,目前有4个gpu,但使用的时候一直显示用的cuda0,指定其他cuda也不生效 #66

Open GUORUIWANG opened 2 years ago