加載預訓練 pytorch model

Question

盡管問題看起來很簡單，但無法在其他地方找到此解決方案。

我有一個 pytorch (.pt) 文件，我正在嘗試加載它。 我知道我需要先構建 model

model = MyModel()

但是我的 pytorch 文件構建了一個 model (se_resnext101_32x4d)，我沒有為它制作 class。 因此，當我嘗試做

model = se_resnext101_32x4d()

我收到一個錯誤

name 'se_resnext101_32x4d' is not defined

我試過做

import pretrainedmodels

model = pretrainedmodels.__dict__[se_resnext101_32x4d]()

但錯誤仍然存在。

Answer 1

經過一番搜索，您似乎正在嘗試使用包含預訓練模型和API的 package 來下載和使用它們。 根據他們的文檔，您可以像這樣加載 model ：

import pretrainedmodels

Model = pretrainedmodels.__dict__['se_resnext101_32x4d']
model = Model(num_classes=1000, pretrained='imagenet')
model.eval()

如果您還沒有安裝pip package，請不要忘記安裝：

pip install pretrainedmodels