Model saved by torch2.1.0 can't load in 2.0.0

These issues are expected since PyTorch is not forwards compatible but should be backwards compatible, i.e. loading an older checkpoint in newer PyTorch versions should work.