查看pth模型

.pth模型文件是PyTorch中保存模型的一种常见格式，可以通过以下代码来加载和查看.pth文件中的模型信息：

import torch

# 加载.pth文件
model = torch.load('model.pth')

# 查看模型结构
print(model)

# 查看模型参数
for name, param in model.named_parameters():
    print(name, param)

这段代码首先使用`torch.load()`函数加载.pth文件中的模型，并将其存储在变量`model`中。然后，通过`print()`函数输出模型的结构信息，可以看到模型的各个层以及它们的参数。最后，使用一个循环遍历模型的所有参数，并打印出每个参数的名称和数值。

需要注意的是，如果你的模型是在GPU上训练的，那么在加载.pth文件时，需要指定`map_location`参数将模型参数映射到CPU上，例如：

model = torch.load('model.pth', map_location=torch.device('cpu'))