torch.save¶
- torch.save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL, _use_new_zipfile_serialization=True)[源代码]¶
将对象保存到磁盘文件。
另请参见:保存和加载张量
- 参数
注意
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)