TypeError: torch.nn.modules.batchnorm.BatchNorm2d is not a Module subclass
torch.nn.modules.batchnorm.BatchNorm2d问题
报错代码:
self.backbone = nn.Sequential(
resnet.conv1,
resnet.norm_layer,
resnet.actn,
resnet.avg_pool,
resnet.layer1,
)
修改为:
self.backbone = nn.Sequential
(
resnet.conv1,
resnet.norm_layer,
resnet.actn,
resnet.avg_pool,
resnet.layer1,
)
正常运行,不再报错。