For reasons, it just I accidentally use torch-xla nightly in a TPU machine and that nightly come along with torch 1.6.
Training ResNet50 with 20 times faster than my GPU, which is great. But then I can’t load the saved weights to make use of it even though I’m using the 1.5.0.
The latest pip and conda I found is also 1.5, I’m not gonna try upgrade my PC lib to compile 1.6 it in risk of breaking something else, happens a lot as you might expect from Ubuntu, thus this situation.
If you want to say “just update it and everything will work”, even though I only need to change one character and it’ll work too.
Question is, what cause more trouble between

Pytorch loading the file has a chance of doesn’t know how to handle some content of the file, even though most of the time nothing’s gonna happen
Raise the error if version mismatch regardless of what’s in the file, even though most Pytorch users never use save() and load() for anything other than some float32