torch.save

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

将对象保存到磁盘文件中。

参见：保存和加载张量

参数:

• obj（对象）- 已保存的对象

• 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 切换到使用新的基于 zip 文件格式的文件格式。 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)