深度学习权重分类

.bin文件是一种二进制文件，用于保存Pytorch模型的权重信息。它只包含了模型的参数数值，没有保存模型的结构信息。因此，当使用.bin文件加载模型时，我们需要已经定义好模型的结构，并与.bin文件中的参数数值相匹配。bin文件的大小较小，加载速度较快，因此在生产环境中使用较多。

import torch
from model import MyModel

# 创建并加载模型
model = MyModel()
model.load_state_dict(torch.load('model_weights.bin'))  # 注意这里是model.load_state_dict

.pt文件是一种推荐的保存Pytorch模型的文件类型，它包含了模型的权重和结构信息。与.bin文件不同，.pt文件可以完整保存模型的所有信息，使得我们可以方便地加载整个模型，而无需额外定义模型结构。

import torch
from model import MyModel

# 保存模型
model = MyModel()
torch.save(model, 'model.pt')

# 加载模型
loaded_model = torch.load('model.pt')   # 注意这里是torch.load