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

Question

我正在以这种方式创建内置火炬模型：

m = torchvision.models.resnet50()

，然后我想使用变量m查找模型的名称（在这种情况下为resnet50）。 致电type(m)时，我得到torchvision.models.resnet.ResNet，这几乎是答案，但深度对我也很重要。我可以尝试迭代m.modules（以这种方式长151个模块）并推断出重新块的深度，但这在其他模型（如inception，alexnet等）中将完全不同。

Answer 1

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

所以唯一的方法就是上面的Umang Gupta建议的方法：写一个包装模型

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