基于BART的长文本生成demo

基于BART的长文本生成demo,第1张

这是一个使用pytorch,pytorch_lightning框架简单基于NLG领域预训练模型BART进行故事生成的demo。BART预训练模型来源于Hugginface: bart-base.

代码可见:
github: story-generation-demohttps://github.com/tangg555/story-generation-demo或者
gitee: story-generation-demohttps://gitee.com/travis-tang/story-generation-demo

运行程序的前置条件:
- Python 3 or Anaconda (mine is 3.8)
- Pytorch 
- transformers (a package for huggingface)
- pytorch-lightning (a package)


欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/723748.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-26
下一篇 2022-04-26

发表评论

登录后才能评论

评论列表(0条)

保存