torch.save

torch.save(obj, f, pickle_module=pickle, pickle_protocol=2, _use_new_zipfile_serialization=True)

將物件儲存到磁碟檔案。

另請參閱：儲存和載入張量

引數

• obj (object) – 要儲存的物件

• f (Union[str, PathLike[str]], IO[bytes]]) – 一個類檔案物件（必須實現 write 和 flush 方法）或包含檔名的字串或 os.PathLike 物件

• pickle_module (Any) – 用於對元資料和物件進行 pickling 的模組

• pickle_protocol (int) – 可以指定以覆蓋預設協議

注意

PyTorch 的一個常用約定是使用 .pt 副檔名儲存張量。

注意

PyTorch 在序列化過程中會保留儲存共享。有關更多詳細資訊，請參閱儲存和載入張量保留檢視。

注意

PyTorch 1.6 版本將 torch.save 切換為使用新的基於 zipfile 的檔案格式。torch.load 仍然保留載入舊格式檔案的能力。如果出於任何原因希望 torch.save 使用舊格式，請傳遞關鍵字引數 _use_new_zipfile_serialization=False。

示例

>>> # Save to file
>>> x = torch.tensor([0, 1, 2, 3, 4])
>>> torch.save(x, "tensor.pt")
>>> # Save to io.BytesIO buffer
>>> buffer = io.BytesIO()
>>> torch.save(x, buffer)