Unfortunately I gave that a try and it doesn’t seem to change the problem. I use .eval() before saving the network (which is done using:
torch.save(model, IMG_PATH+'CategoricalNet.pt')
and then repeat .eval() when loading in the network again just to be sure.
Thanks for the reply by the way!