torch.save

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

將物件儲存到磁碟檔案。

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

參數

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

• f (Union[str, PathLike, BinaryIO, IO[bytes]]) – 類似檔案的物件（必須實作 write 和 flush）或包含檔案名稱的字串或 os.PathLike 物件

• pickle_module (Any) – 用於 pickle 後設資料和物件的模組

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

注意

一個常見的 PyTorch 慣例是使用 .pt 檔案副檔名來儲存張量。

注意

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

注意

PyTorch 的 1.6 版本已切換 torch.save 以使用新的基於 zipfile 的檔案格式。torch.load 仍然保留載入舊格式檔案的能力。如果由於任何原因您希望 torch.save 使用舊格式，請傳遞 kwarg _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)