pth模型保存时是按照“整个模型保存”和“只保存模型参数”会影响模型的加载和访问方式
保存方式为“整个模型”（torch.save(model, PATH)）：
import torch
if __name__ == '__main__':
model_pth = r'D:\${modelPath}\${modelName}.pth'
net = torch.load(model_pth, map_location=torch.device('cpu'))
for key, value in net["state_dict"].items():
print(key,value.size(),sep=" ")输出（部分截图）为：
文章来源：https://www.toymoban.com/news/detail-599784.html
保存方式为“只保存模型参数”（torch.save(model.state_dict(), PATH)）：
待补充文章来源地址https://www.toymoban.com/news/detail-599784.html
到了这里，关于PyTorch：通过pth文件查看网络结构（查看输入输出维度）的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
