torch.save

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

將物件儲存到磁碟檔案。

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

有關操作 checkpoint 的更高階工具，請參閱佈局控制。

引數

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

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

• pickle_module (Any) – 用於醃製元資料和物件的模組

• 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)