前言

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小结的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。