920232796 / bert_seq2seq

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

关于UniLM #65

Open cppww opened 2 years ago

cppww commented 2 years ago

想请问下 使用了UniLM的方案是通过 哪部分代码体现的啊

920232796 commented 2 years ago

seq2seq_model.py里面,forward函数中。