(十三)网络模型的保存与读取

最新推荐文章于 2024-10-02 16:46:15 发布

小羊咩~

最新推荐文章于 2024-10-02 16:46:15 发布

阅读量575

点赞数

分类专栏： PyTorch 文章标签： pytorch 深度学习神经网络

本文链接：https://blog.csdn.net/m0_58702532/article/details/124541081

版权

PyTorch 专栏收录该内容

16 篇文章 6 订阅

订阅专栏

【声明】来源b站视频小土堆PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】_哔哩哔哩_bilibili

import torch
import torch.nn as nn
import torchvision
#参数是没有经过训练的
vgg16 = torchvision.models.vgg16(pretrained=False)

#保存方式1
#既保存网络，又保存了模型参数
torch.save(vgg16,"./data/vgg16.pth")

#保存方式2,只保存了模型参数
torch.save(vgg16.state_dict(),"./data/vgg16_2.pth")

import torch

#保存方式1 加载模型
# model = torch.load("./data/vgg16.pth")
# print(model)

#保存方式2 加载模型
model2 = torch.load("./data/vgg16_2.pth")
print(model2)