【Pytorch】模型保存與加載

Recommended approach for saving a model

There are two main approaches for serializing and restoring a model.

The first (recommended) saves and loads only the model parameters:

torch.save(the_model.state_dict(), PATH)

Then later:

the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

The second saves and loads the entire model:

torch.save(the_model, PATH)

Then later:

the_model = torch.load(PATH)

However in this case, the serialized data is bound to the specific classes and the exact directory structure used, so it can break in various ways when used in other projects, or after some serious refactors.

---

參考鏈接：Recommended approach for saving a model

                  Saving and Loading Models

                  PyTorch 學習筆記（五）：存儲和恢復模型並查看參數

                  Best way to save a trained model in PyTorch?

                  Saving torch models