/gpt2-chatbot-nice

基于transformers的gpt2聊天机器人助手

Primary LanguageJupyter Notebook

gpt2-chatbot-nice

基于transformers的gpt2聊天机器人助手

环境

  • python==3.8
  • torch==1.9
  • transformers==4.24.0

生成的model文件夹是train的每轮训练模型

生成的mim_ppl_model是val_loss低于设定loss的模型

一轮对话机器人,可改造成多轮对话机器人

预测模块还在重构ing

借鉴来源已丢,实在抱歉

其他...(想起来再加!!!)