python - 如何从模型本身找到 torch 模型名称？

Question

我正在以这种方式创建内置 torch 模型:

m = torchvision.models.resnet50()

我想随后使用变量 m 发现模型的名称(在本例中为 resnet50)。当调用 type(m) 时，我得到 torchvision.models.resnet.ResNet 这几乎就是答案，但深度对我来说也很重要。我可以尝试迭代 m.modules (以这种方式有 151 个模块长)并推断 resblocks 深度，但这在其他模型(如 inception、alexnet 等)中的工作方式完全不同。

Answer 1

如果我们查看代码here，ResNet50是一个函数的名称.

所以唯一的方法是上面 Umang Gupta 建议的方法:编写一个 Wrapper 模型

model = torchvision.models.resnet50()
model.name = 'Resnet-50'