PyTorch之模型的保存与加载

最新推荐文章于 2024-05-19 19:02:56 发布

Never-Giveup

最新推荐文章于 2024-05-19 19:02:56 发布

阅读量565

点赞数 1

分类专栏： Pytorch框架文章标签： Pytorch 模型保存模型加载

本文链接：https://blog.csdn.net/qq_36653505/article/details/84728395

版权

本文通过一个小型的演示，详细介绍了在PyTorch中如何保存和加载模型。首先，展示了如何保存整个模型，包括模型结构和权重；然后，解释了仅保存模型参数的方法，这对于迁移学习或继续训练已训练的模型非常有用。

摘要由CSDN通过智能技术生成

从一个小demo讲解

import torch
import matplotlib.pyplot as plt
from torch.autograd import Variable


# 生成数据
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim = 1)
y = x.pow(2) + 0.2 * torch.rand(x.size())

# 变为Variable
x, y = Variable(x), Variable(y)

# 定义网络
net = torch.nn.Sequential(
    torch.nn.Linear(1, 10),
    torch.nn.ReLU(),
    torch.nn.Linear