Where did you tune these parameters? Directly in the source file (in torchvision) or in another script?
I don’t think that torch.save(model, path) will store all source files with it, and would thus try to load them in torch.load.
Note that I’m speculating a lot here, as I’m not using this workflow, since it can break in many ways.