文章目录

• • 保存模型
  • • 方法一：只保存模型参数
    • 方法二：保存整个模型
  • 加载模型
  • • 方法一：加载模型参数（推荐）
    • 方法二：加载整个模型
  • 训练时保存模型
  • 加载并继续训练
  • 总结

保存模型

在PyTorch中，保存模型有两种常见的方法：

方法一：只保存模型参数

这种方法只保存模型的参数（weights和biases），文件较小且灵活性更高。

import torch

# 假设有一个模型实例
model = ...

# 保存模型参数
torch.save(model.state_dict(), 'model.pth')
</