Kaixhin:

model.eval()


Kai,
This worked for me also. Finding this solution was way too hard. I did not see this in the documentation anywhere. I searched every which way until your solution came up. The PyTorch Models documentation needs to explain this. I could get things working in Keras but not Resnet50 in PyTorch.
Do I need to use the same model.eval() when I load a customer model based on ResNet50 with tuning or is this unique to the ImageNet version?
Thank you!