#176 · 主分类: AI对话与Chatbot

GPT2-chitchat

chichat dialogpt dialogue-model gpt-2 gpt2 nlp text-generation transformer

用于中文闲聊的GPT2模型

项目最后更新:10/30/23

GitHub Stars

3.0K

Forks数量

665

贡献者数量

1

许可证

Other

收录理由

这是一个基于GPT2的中文闲聊机器人项目,代码仓库里把数据预处理、模型训练和命令行交互聊天都放在了一起,拿到手就能从原始语料一路跑到能对话的模型,省去自己拼装各环节的麻烦。生成回复时用到了温度采样、Top-k采样和Nucleus采样,能有效避免模型陷入重复输出的死循环。代码里附带大量中文注释,讲得很细,对想搞懂多轮对话微调流程的人来说是很好的学习材料;作者也把训练用的闲聊数据集一并公开了。微软的DialoGPT项目引用过这个工作,拿来当原型验证想法是够用的。不过依赖版本偏老,锁在PyTorch 1.7和transformers 4.2上,真要上生产环境还得自己动手升级改造。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目