负荷预训练pytorch模型

Question

虽然这个问题看起来很简单，但我们还没有在其他地方找到这个解决方案

我有一个pytorch（.pt）文件，正在尝试加载它。我知道我需要首先通过以下方式构建模型

model = MyModel()

但是我的pytorch文件构建了一个模型（se_resnext101_32x4d），我没有为它创建类。当我试着去做的时候

model = se_resnext101_32x4d()

我犯了一个错误

name 'se_resnext101_32x4d' is not defined

我试过了

import pretrainedmodels

model = pretrainedmodels.__dict__[se_resnext101_32x4d]()

但错误依然存在

Answer 1

经过一点搜索，您似乎正在尝试使用this package，它包含预训练模型和一个API来下载和使用它们。根据their documentation，您可以加载如下模型：

import pretrainedmodels

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

如果您还没有安装pip软件包，请不要忘记：

pip install pretrainedmodels