无法从 torchvision.models.resnet 导入“BasicBlock”

Question

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

然后在我的代码中使用它

为什么我会收到此错误？

Answer 1

BasicBlock确实已定义，但模块并未导出它：请参阅此处的定义__all__。所以torchvision/models/resnet.py只导出这些：ResNet, resnet18, resnet34, resnet50, resnet101, resnet152, resnext50_32x4d, resnext101_32x8d, wide_resnet50_2, 和wide_resnet101_2。