15 pytorch版本问题Unexpected running stats buffer(s) "initial_layers.1.running_mean" and "initial_lay...

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)

你可能感兴趣的:(15 pytorch版本问题Unexpected running stats buffer(s) "initial_layers.1.running_mean" and "initial_lay...)