当对现有模型使用print时,它不会打印模型。相反,它显示如下内容: <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’ object has no attribute ‘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层和模型拓扑吗?
回答:
models.resnext101_32x8d
是类构造函数,你需要调用构造函数,只需在末尾添加括号即可。
m1 = models.resnext101_32x8d()print(m1)