ckpt和pth的文件 使用python 怎么使用

"ckpt"和"pth"通常是在PyTorch中用于保存模型权重的文件格式。这两个术语可以代表检查点（Checkpointing）文件。

1. **ckpt**：这是TensorFlow早期使用的文件格式，但现在更推荐使用PyTorch的.pth格式。虽然有些库支持转换，但在PyTorch中直接创建.pth文件更为常见。

2. **pth**：PyTorch的标准模型存档格式，`.pth`文件包含了模型的参数、优化器状态和其他相关信息。你可以使用`torch.save()`函数来保存模型，如下所示：

   import torch

   # 假设model是一个PyTorch模型，optimizer是训练它的优化器
   torch.save(model.state_dict(), 'model.pth')

如果你想从.pth文件恢复模型，可以这样做：

   model = YourModel()  # 初始化一个新模型
   model.load_state_dict(torch.load('model.pth'))