时间戳

首页 > 解决方案 > 如何查看通过加载 .pth 文件设置的模型的前向功能？

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

问题描述

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

标签： pytorch

解决方案

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

import inspect
import torch

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

推荐阅读