# 原始代码
nn.AdaptiveAvgPool2d(1),
nn.Conv2d(channels, out_channels, kernel_size=1, stride=1, padding=0),
nn.InstanceNorm2d(out_channels)
特征经过AdaptiveAvgPool2d(1)后,变成维度1*1的特征,无法通过InstanceNorm2d进行正则化,可以将其改成BatchNorm2d,如下:
AdaptiveAvgPool2d(1)
nn.Conv2d(channels, out_channels, kernel_size=1, stride=1, padding=0),
nn.BatchNorm2d(out_channels)