https://blog.csdn.net/bcfd_yundou/article/details/90295328
norm_layer = functools.partial(nn.InstanceNorm2d, affine=False)
修改为:
norm_layer = functools.partial(nn.InstanceNorm2d, affine=False, track_running_stats=True)
https://blog.csdn.net/bcfd_yundou/article/details/90295328
norm_layer = functools.partial(nn.InstanceNorm2d, affine=False)
修改为:
norm_layer = functools.partial(nn.InstanceNorm2d, affine=False, track_running_stats=True)