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

Posted 2023-03-12

【中文标题】在 pytorch 中为聊天机器人加载经过训练的模型保存

【英文标题】：Load the ,trained model save, in pytorch for chatbots

【发布时间】：2021-07-11 14:16:25

【问题描述】：

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

【参考方案1】：

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

# 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。