Получение ошибки при использовании print () или summary () в pytorch для просмотра слоев и размеров веса в модели Pytorch

Question

При использовании печати на существующей модели модель не печатается. Вместо этого он показывает: <function resnext101_32x8d at 0x00000178CC26BA68>

>>> import torch
>>> import torchvision.models as models 
>>> m1 = models.resnext101_32x8d
>>> print(m1)
<function resnext101_32x8d at 0x00000178CC26BA68>
>>>

При использовании summary выдает следующую ошибку:

AttributeError: у объекта 'function' нет атрибута 'apply'

>>> import torch
>>> import torchvision.models as models 
>>> from torchvision import summary
>>> m1 = models.resnext101_32x8d
>>>
>>> summary(m1, (3, 224, 224))
 Traceback(most recent call last):
   File "<stdin>", line 1, in <module>
   File torchsummary.py, line 68, in summary
      model.apply(register_hook)
AttributeError: 'function' object has no attribute 'apply'

Как исправить эти проблемы, связанные с print и summary? Есть ли другие способы легко увидеть все слои Pytorch и топологию модели?

Answer 1

models.resnext101_32x8d это конструктор класса, вам нужно вызвать конструктор, просто добавьте скобки в конце.

m1 = models.resnext101_32x8d()
print(m1)