如何使用自定义数据集训练BART进行文本汇总?

时间:2020-04-23 14:56:38

标签: python nlp pytorch summarization huggingface-transformers

我已经准备了一个自定义数据集,用于训练自己的自定义模型进行文本汇总。我希望使用BART,因为它是目前的最新水平。我正在使用pytorch使用HuggingFace的Transformer库。

但是我不知道如何自己训练模型,我尝试在以下链接上运行run_train.sh: https://github.com/huggingface/transformers/tree/master/examples/summarization/bart

我已将CNN数据加载到Google Colab中。仍然显示错误。

有人可以帮助我如何使用转移学习进行自定义文本摘要吗? 任何其他模型也有帮助。

0 个答案:

没有答案