920232796 / bert_seq2seq

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

您好,我想添加一些我自己的数据集做fine-tune,请问应该怎么做呢

pangsg opened this issue · comments

您好,我想添加一些我自己的数据集做fine-tune,请问应该怎么做呢

先看懂某个例子,知道应该传入给模型什么输入跟输出,然后读取你自己的数据集,构建好输入输出就好了,剩下的就不用管了。