PyTorch 序列化語義

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

最佳實(shí)務(wù)

推薦的模型保存方法

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

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

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

然后再：

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

第二個(gè)保存并加載整個(gè)模型：

torch.save(the_model, PATH)

Then later:

the_model = torch.load(PATH)

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