加载预训练 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