torch.save

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

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

另请参阅: 保存和加载张量

Parameters

• obj (对象) – 保存的对象

• f (联合[字符串, 路径类, 二进制IO, IO[字节]]) – 一个类文件对象（必须实现write和flush）或一个包含文件名的字符串或os.PathLike对象

• pickle_module (任意) – 用于序列化元数据和对象的模块

• pickle_protocol (int) – 可以指定以覆盖默认协议

注意

PyTorch 的一个常见惯例是使用 .pt 文件扩展名保存张量。

注意

PyTorch 在序列化过程中保留存储共享。更多详情请参见 保存和加载张量保留视图。

注意

PyTorch 1.6 版本将 torch.save 切换为使用基于 zip 文件的新文件格式。torch.load 仍然保留了加载旧格式文件的能力。如果由于任何原因您希望 torch.save 使用旧格式，请传递关键字参数 _use_new_zipfile_serialization=False。

示例

>>> # 保存到文件
>>> x = torch.tensor([0, 1, 2, 3, 4])
>>> torch.save(x, 'tensor.pt')
>>> # 保存到 io.BytesIO 缓冲区
>>> buffer = io.BytesIO()
>>> torch.save(x, buffer)