Pytorch 如何查看通过加载.pth文件设置的模型的转发功能？

我有一个.pth文件，我可以使用

model=torch.load（“path to.pth”）

---

但我不知道它是如何向前发展的。有没有办法检查模型的正向功能？Thx.

我假设您的

.pth

文件包含整个转储模型，而不仅仅是state dicts，这就是您陷入这种困境的原因。但在任何情况下，都可以使用

inspect

库实际检查forward函数

import inspect
import torch

unknown_model = torch.load('save_path.pth')
print(inspect.getsource(unknown_model.forward))

---

我假设您的

.pth

文件包含整个转储模型，而不仅仅是state dicts，这就是您陷入这种困境的原因。但在任何情况下，都可以使用

inspect

库实际检查forward函数

import inspect
import torch

unknown_model = torch.load('save_path.pth')
print(inspect.getsource(unknown_model.forward))

---

您的模型定义正确吗？否则，您首先如何实例化模型？你可以在你的模型定义中检查向前传球。你的模型定义正确吗？否则，您首先如何实例化模型？然后，可以在模型定义中检查前向传递。