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

transfer-learning-conv-ai

chatbots deep-learning dialog gpt gpt-2 neural-networks nlp pytorch transfer-learning

🦄 最先进的对话式AI,采用迁移学习

项目最后更新:06/12/23

GitHub Stars

1.8K

Forks数量

432

贡献者数量

13

许可证

MIT

收录理由

该仓库将 Hugging Face 参加 ConvAI2 竞赛的代码浓缩成约 250 行带注释的训练脚本,对于想学习如何基于预训练 GPT 或 GPT-2 微调对话模型的人来说,是一份很实用的参考材料。代码从训练到推理都有清晰的注释,涵盖了分布式训练和 FP16 精度选项,并附带交互脚本,可以直接加载官方发布的模型权重,立刻和机器人聊天。如果想自己重新训练一个版本,在 8 卡 V100 集群上大约只需要一个小时,这对小团队快速验证想法来说成本可控。虽然这套代码更多是研究性质的示例,而非持续维护的生产级聊天机器人,但它仍然是理解迁移学习如何应用于对话系统的良好起点。

相关文章

暂无针对该项目的文章。

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

同类热门项目