深度学习中的.pt和.pth文件的区别

什么是.pt和.pth文件？

首先，.pt和.pth文件都是用于保存PyTorch模型的文件格式。它们可以用于保存模型的权重、结构以及其他相关信息。

.pt文件

.pt文件是PyTorch模型保存的通用格式。通常，.pt文件可以保存以下内容：

1. 模型的权重（weights）：这是最常见的用法，只保存模型的参数。
2. 完整的模型（full model）：包括模型的架构和权重。在这种情况下，可以通过简单的一行代码来加载整个模型。

保存模型权重代码

import torch

# 假设有一个训练好的模型 model
torch.save(model.state_dict(), 'model_weights.pt')

保存完整模型代码