快捷方式

torch.save

torch.save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL, _use_new_zipfile_serialization=True)[源代码]

将对象保存到磁盘文件。

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

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

  • f (联合[字符串, PathLike, BinaryIO, IO[字节]]) – 类文件对象(必须实现写入和刷新)或包含文件名的字符串或 os.PathLike 对象

  • pickle_module (任何) – 用于对元数据和对象进行腌制的模块

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

注意

PyTorch 的常见约定是使用 .pt 文件扩展名保存张量。

注意

PyTorch 在序列化中保留存储共享。有关更多详细信息,请参阅 保存和加载张量保留视图

注意

PyTorch 1.6 版本将 torch.save 切换为使用新的基于压缩包的文件格式。 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)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源