前言

PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是：

torchvision.datasets torchvision.models torchvision.transforms

笔记

官网介绍 源码

---

本文介绍 torchvision.models 如何使用。以 vgg16为例子

导入预训练模型：

import torchvisionmodel = torchvision.models.vgg16(pretrained=True)

只导入网络结构，不导入参数：

model = torchvision.models.vgg16(pretrained=False) #主要是这里改为False

由于 pretrained 参数默认是 False，所以 2) 等价于：

model = torchvision.models.vgg16()

————————————————
版权声明：本文为CSDN博主「梦dancing」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_15192373/article/details/83658005

参考：

模型调整
pretrained模型修改
torch.models

总结

以上是生活随笔为你收集整理的PyTorch torchvision.models小结的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。