skyerhxx / Chatbot

基于pytorch的聊天机器人项目

Chatbot

基于pytorch的聊天机器人项目

开发环境

• Pytorch1.3
• Python3.7
• cuda10.1
• Win10

数据集

• 青云数据集

文件说明&运行方式

• train.py是用来训练model.pkl的 训练得到train.pkl，在运行demo.py的时候使用
• 运行方式 运行demo.py python demo.py --model model.pkl --device cuda

文件应该还有model.pkl(modelA.pkl, modelB.pkl)

因为文件较大，没有上传