Detailed Explanation of Model Checkpoint
Saving weights alone differs from saving a checkpoint that can resume training. PyTorch examples also store optimizer state and training progress for the latter. Check which architecture and software version the file expects.
Source: Official documentation.