前言

PyTorch框架中有一個非常重要且好用的包：torchvision，該包主要由3個子包組成，分別是：

torchvision.datasets torchvision.models torchvision.transforms

筆記

官網(wǎng)介紹 源碼

---

本文介紹 torchvision.models 如何使用。以 vgg16為例子

導(dǎo)入預(yù)訓(xùn)練模型：

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

只導(dǎo)入網(wǎng)絡(luò)結(jié)構(gòu)，不導(dǎo)入?yún)?shù)：

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

由于 pretrained 參數(shù)默認(rèn)是 False，所以 2) 等價于：

model = torchvision.models.vgg16()

————————————————
版權(quán)聲明：本文為CSDN博主「夢dancing」的原創(chuàng)文章，遵循 CC 4.0 BY-SA 版權(quán)協(xié)議，轉(zhuǎn)載請附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/qq_15192373/article/details/83658005

參考：

模型調(diào)整
pretrained模型修改
torch.models

總結(jié)

以上是生活随笔為你收集整理的PyTorch torchvision.models小结的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。