我已经准备了一个自定义数据集,用于训练自己的自定义模型进行文本汇总。我希望使用BART,因为它是目前的最新水平。我正在使用pytorch使用HuggingFace的Transformer库。
但是我不知道如何自己训练模型,我尝试在以下链接上运行run_train.sh: https://github.com/huggingface/transformers/tree/master/examples/summarization/bart
我已将CNN数据加载到Google Colab中。仍然显示错误。
有人可以帮助我如何使用转移学习进行自定义文本摘要吗? 任何其他模型也有帮助。