如何将pytorch模型保存为.pt文件

您可以使用以下代码将 PyTorch 模型保存为 .pt 文件：

import torch

# 定义模型
model = YourModel()

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

其中，`YourModel()` 是您定义的 PyTorch 模型，`model.state_dict()` 是模型的参数字典，`'model.pt'` 是保存模型的文件名。