在聊天机器人的 pytorch 中加载训练好的模型保存

Question

我运行了本教程的代码 (link)，经过一段时间学习模型完成后，我与经过训练的模型聊天。退出程序后，在程序文件旁边的save目录下，创建了8个扩展名为tar的文件。 我猜他们是经过训练的模型保存文件，我应该怎么做才能加载这些文件并重用它们？

Answer 1

在“Run the Model”中，它们描述了您应该做什么。更具体地说，他们的代码段中有这段代码：

# Set checkpoint to load from; set to None if starting from scratch
loadFilename = None
checkpoint_iter = 4000
#loadFilename = os.path.join(save_dir, model_name, corpus_name,
#                            '{}-{}_{}'.format(encoder_n_layers, decoder_n_layers, hidden_size),
#                            '{}_checkpoint.tar'.format(checkpoint_iter))

并且您基本上需要取消注释 #loadFilename 并将 checkpoint_iter 设置为您想要的迭代。之后，您可以跳过训练部分，因为您已经运行了它，然后再次运行 evaluation code。