深入了解 torchvision.models.resnet18 新舊版本的區(qū)別

在深度學(xué)習(xí)模型開發(fā)中，PyTorch 和 torchvision 一直是我們不可或缺的工具。近期，torchvision 對其模型加載 API 進行了更新，將舊版的 pretrained 參數(shù)替換為新的 weights 參數(shù)。本文將介紹這一變化的背景、具體區(qū)別，以及如何在代碼中適配新版本的 torchvision.models.resnet18。

一、背景

隨著深度學(xué)習(xí)模型的不斷發(fā)展，torchvision 的模型庫中預(yù)訓(xùn)練權(quán)重的種類逐漸增多。為了更靈活、更明確地管理這些權(quán)重，開發(fā)者決定將 pretrained 參數(shù)替換為更具描述性的 weights 參數(shù)。

• 舊版：pretrained 參數(shù)

  • 僅支持布爾值：True 表示加載默認的預(yù)訓(xùn)練權(quán)重，False 表示不加載預(yù)訓(xùn)練權(quán)重。
  • 模糊：沒有明確說明具體加載了哪個版本的權(quán)重。

• 新版：weights 參數(shù)</