search

920232796 / bert_seq2seq

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

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

Open GUORUIWANG opened 1 year ago