# 序列化語義
# 序列化語義
## 最佳實踐
### 保存模型的推薦方法
這主要有兩種方法序列化和恢復模型。
第一種(推薦)只保存和加載模型參數:
```
torch.save(the_model.state_dict(), PATH)
```
然后:
```
the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))
```
第二種保存和加載整個模型:
```
torch.save(the_model, PATH)
```
然后:
```
the_model = torch.load(PATH)
```
然而,在這種情況下,序列化的數據被綁定到特定的類和固定的目錄結構,所以當在其他項目中使用時,或者在一些嚴重的重構器之后它可能會以各種方式break。
- PyTorch 中文文檔
- 主頁
- 自動求導機制
- CUDA語義
- 擴展PyTorch
- 多進程最佳實踐
- 序列化語義
- torch
- torch.Tensor
- torch.Storage
- torch.nn
- torch.nn.functional
- torch.autograd
- torch.optim
- torch.nn.init
- torch.multiprocessing
- torch.legacy
- torch.cuda
- torch.utils.ffi
- torch.utils.data
- torch.utils.model_zoo
- torchvision
- torchvision.datasets
- torchvision.models
- torchvision.transforms
- torchvision.utils
- 致謝