machine-learning - Pytorch 使用 argprse 传递架构类型

Question

使用 Pytorch。使用以下代码传递架构类型时:

parser.add_argument('-arch', action='store',
                    dest='arch',
                    default= str('vgg16'))

通过以下代码使用架构名称时:

model = models.__dict__['{!r}'.format(results.arch)](pretrained=True)

我收到以下错误:

model = models.dict'{!r}'.format(results.arch) KeyError: "'vgg16'"

我做错了什么？

Answer 1

你有KeyError这意味着您导入的模型不包含'vgg16'作为已知模型之一。
通过打印检查您拥有哪些型号

print(models.__dict__.keys())

这应该可以让您知道导入了哪些模型以及缺少哪些模型，然后您可以查看导入并查看 'vgg16' 在哪里丢失。