Sorry, I am using a new computer and I just installed the newest version of Pytorch.
My code runs perfectly on my old computer.
But now BatchNorm2d is giving me this error when the model is switched to evaluation mode.
‘BatchNorm2d’ object has no attribute ‘track_running_stats’
Does anyone know how to solve this?
I saw the source code change warning, but I am not sure about how to solve it.
My model is based on pre-trained ResNet101