小编김수호*_*김수호的帖子

我正在寻找一种方法来保存 pytorch 模型，并在没有模型定义的情况下加载它。我的意思是我想保存我的模型，包括模型定义。

例如，我想要两个脚本。第一个将定义、训练和保存模型。第二个将加载和预测模型而不包括模型定义。

使用的方法torch.save(), torch.load()要求我在预测脚本中包含模型定义，但我想找到一种方法来加载模型而不在脚本中重新定义它。