Things on this page are fragmentary and immature notes/thoughts of the author. Please read with your own judgement!
-
PyTorch uses pickle to serialize and deserialize objects.
-
The PyTorch convention is to use the file extension
.pt
or.pth
for saving model (or its parameters) and use the file extension.tar
for saving checkpoints. -
It is preferred to save model parameters rather than the whole model.