Как узнать название модели горелки из самой модели?

Question

Я создаю модель встроенного резака следующим образом:

m = torchvision.models.resnet50()

и хочу узнать название модели (в данном случае resnet50), используя впоследствии переменную m.При звонке type(m) я получаю torchvision.models.resnet.ResNet, что почти ответ, но глубина также важна для меня.Я мог бы попытаться перебрать m.modules (таким образом, это 151 модуль длиной) и вывести глубину повторной блокировки, но в других моделях это будет работать совсем по-другому (например, в начале, alexnet и т. Д.).

Answer 1

ResNet50 - это имя функции, если мы посмотрим на код здесь

Таким образом, единственный способ, предложенный Умангом Гуптой выше: написать модель Wrapper

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