PyTorch torchvision.models小结

前言

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

torchvision.datasets
torchvision.models
torchvision.transforms

笔记

官网介绍 源码

---

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

1. 导入预训练模型：

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

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

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

3. 由于 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