pytorch 模型的后缀名

在PyTorch中，模型文件通常使用`.pt`（也称为`.pth`）作为后缀名。这种格式的文件是PyTorch专有的模型保存格式，用于存储训练好的模型参数和结构。当你保存一个PyTorch模型时，可以使用`torch.save()`函数，例如：

model = ... # 初始化模型
torch.save(model.state_dict(), 'model_name.pt')

同样，当你加载模型时，可以使用`torch.load()`配合模型类来恢复模型：

model = YourModelClass()  # 假设YourModelClass是你的模型类
model.load_state_dict(torch.load('model_name.pt'))