Errors when running test.py

[LXX230]

RuntimeError: Error(s) in loading state_dict for ImageTransformNet:
Unexpected running stats buffer(s) "downsamples.0.block.2.running_mean" and "downsamples.0.block.2.running_var" for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved before 0.4.0, this may be expected because InstanceNorm2d does not track running stats by default since 0.4.0. Please remove these keys from state_dict. If the running stats are actually needed, instead set track_running_stats=True in InstanceNorm2d to enable them. See the documentation of InstanceNorm2d for details.
......etc.
Unexpected running stats buffer(s) "residuals.0.block.0.block.2.running_mean" and "residuals.0.block.0.block.2.running_var" for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved before 0.4.0, this may be expected because InstanceNorm2d does not track running stats by default since 0.4.0. Please remove these keys from state_dict. If the running stats are actually needed, instead set track_running_stats=True in InstanceNorm2d to enable them. See the documentation of InstanceNorm2d for details.
......etc.

I inquired about the problem on the Internet. The result said that it was caused by different training using GPU or CPU, but I still couldn't solve it by modifying it. I hope you can communicate, thank you