Closed swoook closed 3 years ago
torch.nn.BatchNorm
requires an input larger than (1, 1) when training self.scale4 = nn.Sequential(nn.AdaptiveAvgPool2d((1, 1)),
BatchNorm2d(inplanes, momentum=bn_mom),
nn.ReLU(inplace=True),
nn.Conv2d(inplanes, branch_planes, kernel_size=1, bias=False),
)
BatchNorm
will always get input size of (1, 1)BatchNorm
is applicable although the input size of it is (1, 1)
Issue description
Code example
System Info