lc222 / seq2seq_chatbot_new

基于seq2seq模型的简单对话系统的tf实现,具有embedding、attention、beam_search等功能,数据集是Cornell Movie Dialogs
143 stars 64 forks source link

加载预训练模型的小bug #20

Open aguang1201 opened 5 years ago

aguang1201 commented 5 years ago

train.py的29行:model.restore(sess, ckpt.model_checkpoint_path) 应该改成:model.saver.restore(sess, ckpt.model_checkpoint_path)