我如何查看通过加载.pth文件设置的模型的前向功能？

Question

我有一个.pth文件，我可以简单地使用model = torch.load("path to .pth")建立网络 但是我不知道它如何转发。有什么方法可以检查模型的正向功能吗？谢谢。

Answer 1

我假设您的.pth文件包含整个转储的模型，而不是仅包含状态命令，这就是您处于这种困境的原因。但是无论如何，是的，您实际上可以使用inspect库来检查转发功能。

import inspect
import torch

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