torch.save

参考 torch.save - 云+社区 - 腾讯云

torch.save(obj, f, pickle_module=<module 'pickle' from '/opt/conda/lib/python3.6/pickle.py'>, pickle_protocol=2, _use_new_zipfile_serialization=False)[source]

讲对象保存为磁盘文件。

See also: Recommended approach for saving a model

参数：

• obj – 保存对象

• f – a file-like object (has to implement write and flush) or a string containing a file name

• f – 类似文件的对象(必须实现写入和刷新)或包含文件名的字符串

• pickle_module – 用来pickling元数据和对象的模块

• pickle_protocol – 是否可以指定重写默认协议

注意

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

警告：

如果你使用Python 2， torch.save()不支持StringIO.StringIO作为有效的类似文件的对象。这是因为写方法应该返回bytes写的数量，StringIO.write()不做这个。请使用io.BytesIO作为替代。

例

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