无法使用torch.load加载保存下来的model

[yysirs]

Trainer训练完成后，使用torch.load，无法加载保存下来的模型
报错:

ModuleNotFoundError: No module named 'model'

我查看了Fastnlp中Trainer保存best model的代码，是

model.cpu()
torch.save(model,model_path)

使用torch.load应该没有问题的，请问作者大大怎么加载训练完成的best model

[houyuchao]

Trainer训练完成后，使用torch.load，无法加载保存下来的模型 报错:

ModuleNotFoundError: No module named 'model'

我查看了Fastnlp中Trainer保存best model的代码，是

model.cpu()
torch.save(model,model_path)

使用torch.load应该没有问题的，请问作者大大怎么加载训练完成的best model

同问

[JustinPeanut]

我在加载模型的时候报错模型对象不可迭代，请问有大佬知道原因吗？