pytorch快速載入預訓練模型引數的方式

2022-01-11 01:52:43 字數 467 閱讀 9692

針對的預訓練模型是通用的模型,也可以是自定義模型,大多是vgg16 ,  resnet50 , resnet101 , 等,從官網載入太慢

通過查詢自己**裡所呼叫網路的類,使用pycharm自帶的函式查詢功能(ctrl+滑鼠左鍵),檢視此網路的載入方法,修改model.load_state_dict()函式。

在本使用者目錄下,linux和win有不同

cd .cache/torch/checkpoints
cd /home/team/.torch/models
兩種方式,常常是用第二種作為torch模型的快取資料夾, /home/team/是使用者的資料夾
進入資料夾把所需模型權重放入即可自動載入,相比第一種方法簡單點。

pytorch 載入預訓練模型

pytorch的torchvision中給出了很多經典的預訓練模型,模型的引數和權重都是在imagenet資料集上訓練好的 載入模型 方法一 直接使用預訓練模型中的引數 import torchvision.models as models model models.resnet18 pretrai...

pytorch載入預訓練模型後,訓練指定層

1 有了已經訓練好的模型引數,對這個模型的某些層做了改變,如何利用這些訓練好的模型引數繼續訓練 pretrained params torch.load pretrained model model the new model model.load state dict pretrained par...

pytorch載入預訓練模型後,訓練指定層

1 有了已經訓練好的模型引數,對這個模型的某些層做了改變,如何利用這些訓練好的模型引數繼續訓練 pretrained params torch.load pretrained model model the new model model.load state dict pretrained par...