pytorch学习笔记(七)---网络模型的使用修改、保存以及加载

最新推荐文章于 2022-12-09 20:17:22 发布

橘春十三99

最新推荐文章于 2022-12-09 20:17:22 发布

阅读量1.5k

点赞数

分类专栏： pytorch 学习笔记文章标签： python pytorch 深度学习

本文链接：https://blog.csdn.net/qq_41858089/article/details/121626983

版权

本篇自学笔记来自于b站《PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】》，Up主讲的非常通俗易懂，文章下方有视频连接，如有需要可移步up主讲解视频，如有侵权，实非故意，深表歉意，请与我联系，删除相关内容！

本节将介绍的内容有：1.使用torchvison中定义好的模型，2.如何修改定义好的模型，3.保存模型的方式和对应的加载模型的方式。

1.使用定义好的模型（以VGG16为例）

给出官方解释：如图可以看到有两个字段，分别为pretrained和progress，pretrained表示是否使用预训练好的模型，该模型是在ImageNet上训练好的。progress则为是否显示下载进度条。

代码为：分别写了pretrained为true和false的两种情况。

vgg16_false= torchvision.models.vgg16(pretrained=False)
vgg16_true = torchvision.models.vgg16(pretrained=

关注