python-3.x - 无法从 torchvision.models.resnet 导入 "BasicBlock"

Question

我正在尝试通过这样做从 torchvision.models.resnet 导入类 BasicBlock

from torchvision.models.resnet import *

它没有给出任何错误，但是当我尝试在我的代码中使用 BasicBlock 类(应该已经导入)时，我得到的错误是

NameError:名称“BasicBlock”未定义

即使 BasicBlock 存在于 torchvision.models.resnet 中

但是当我像这样导入时它没有报错

从 torchvision.models.resnet 导入 BasicBlock

然后在我的代码中使用它

为什么会出现此错误？

Answer 1

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