在新的 python 文件 (PyTorch) 中加载神经网络时遇到问题

Question

我在将经过训练的神经网络加载到新文件时遇到问题。我已经在 Model.py 中保存了神经网络

torch.save(model.state_dict(), "model.pth")

现在在一个新文件（Test.py）中，我有以下代码：

from Model import NeuralNetwork
import torch

model = NeuralNetwork()
model.load_state_dict(torch.load(model.pth))
model.eval()

但是，当我运行这个文件 (Test.py) 时，神经网络似乎在重新训练（我可以看到再次打印的时期）。我很确定这不应该发生。如果有人可以请回复我关于问题是什么，那将是惊人的！提前谢谢！

Answer 1

model.eval() 正在将模型设置为推理模式，我想您需要将一些输入传递给模型才能看到模型的输出。