Pytorch学习（七）---- 保存提取

最新推荐文章于 2024-11-08 20:27:36 发布

原创

最新推荐文章于 2024-11-08 20:27:36 发布 · 265 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#神经网络 #python

本文记录了作者在学习PyTorch过程中关于模型保存的实践，并通过调整学习率，从0.5到0.25~0.3之间找到了最佳的学习效率区间，从而提升了神经网络的训练效果。

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578

import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import os
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
# import os 这里是为了防止报错加的

torch.manual_seed(1)         # reproducible

# fake data
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y = x.pow(2) + 0.2*torch.rand(x.size())
x, y = Variable(x, requires_grad=False), Variable(y, requires_grad=False)

# 神经网络的保存
def save():
    # save net1
    net1 = torch.nn.Sequential(