search

zhaoyingjun / chatbot

ChatGPT带火了聊天机器人，主流的趋势都调整到了GPT类模式，本项目也与时俱进，会在近期更新GPT类版本。基于本项目和自己的语料可以训练出自己想要的聊天机器人，用于智能客服、在线问答、闲聊等场景。
3.51k stars 1.02k forks source link

完成训练时候save 档案出现问题 #156

Open kirinhk opened 1 year ago

kirinhk commented 1 year ago 
Exception has occurred: RuntimeError
[enforce fail at ..\caffe2\serialize\inline_container.cc:365] . invalid file name: model_data\.pt
  File "D:\ai\chatbot-master\Chatbot_pytorch\Seq2seqchatbot\execute.py", line 114, in train
    torch.save({'modelA_state_dict': encoder.state_dict(),
  File "D:\ai\chatbot-master\Chatbot_pytorch\Seq2seqchatbot\execute.py", line 165, in <module>
    train()
Lin-cell-lab commented 8 months ago

你要自己写一个本地路径就可以