如何从模型本身发现火炬型号名称？

Question

我以这种方式创建一个内置的火炬模型：

m = torchvision.models.resnet50()

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

Answer 1

如果我们查看代码here，ResNet50是函数的名称。你知道吗

因此，唯一的方法就是上面Umang Gupta建议的方法：编写一个包装器模型

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