诗词test的程序运行时报错

920232796 / bert_seq2seq

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

Apache License 2.0

1.28k stars 208 forks source link

Open jielingao opened 3 years ago

jielingao commented 3 years ago

读取训练后的roberta模型

920232796 commented 3 years ago

别精简词表

920232796 commented 3 years ago

如果精简的话，会返回两个参数，word2ix 和 keepword。也可以把这里改一下。