pth是啥格式

.pth是PyTorch专用的模型保存格式，它实际上是一个二进制文件，不能直接用文本编辑器打开。下面是一种查看和使用.pth文件的方法：

# 1. 使用PyTorch加载查看
import torch

# 加载模型文件
checkpoint = torch.load('checkpoints/best_model.pth')

# 查看文件内容
print("文件包含的键:", checkpoint.keys())
# 通常会输出: 
# ['model_state_dict', 'optimizer_state_dict', 'epoch', 'best_acc']

# 查看具体参数
print("训练轮次:", checkpoint['epoch'])
print("最佳准确率:", checkpoint['best_acc'])

# 2. 查看模型参数
for name, param in checkpoint['model_state_dict'].items():
    print(f"层名称: {name}")
    print(f"参数形状: {param.shape}")
    print(f"参数类型: {param.dtype}")
    print("---")

注意事项：

• .pth文件只能用PyTorch相关工具打开
• 不要用记事本等文本编辑器打开，会显示乱码
• 文件包含了模型的权重参数和其他训练状态
• 想要进行可视化模型结构的话可以使用：

• • TensorBoard
  • Netron (Netron)
  • PyTorch的模型可视化工具