我试图导入类BasicBlock
从torchvision.models.resnet
做这个
from torchvision.models.resnet import *
它没有错误,但是当我尝试BasicBlock
在代码中使用类(应该已经导入)时,出现了以下错误:
NameError: name 'BasicBlock' is not defined
即使BasicBlock
存在于torchvision.models.resnet
但是当我这样导入时它没有错误
from torchvision.models.resnet import BasicBlock
然后在我的代码中使用它
为什么会出现此错误?
BasicBlock
确实已定义,但是不是模块导出的:请参见此处的定义__all__
。所以torchvision /模型/ resnet.py仅导出这些:ResNet
,resnet18
,resnet34
,resnet50
,resnet101
,resnet152
,resnext50_32x4d
,resnext101_32x8d
,wide_resnet50_2
,和wide_resnet101_2
。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句