PyTorch 序列化語義

原文： https://pytorch.org/docs/stable/notes/serialization.html

最佳實務

推薦的模型保存方法

序列化和還原模型有兩種主要方法。

第一個(推薦）僅保存和加載模型參數(shù)：

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)

Then later:

the_model = torch.load(PATH)

但是，在這種情況下，序列化的數(shù)據(jù)將綁定到所使用的特定類和確切的目錄結構，因此在其他項目中使用時或經過一些嚴重的重構后，序列化的數(shù)據(jù)可能會以各種方式中斷。