[torch 参数更多 ]torch.save

torch.save

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

paddle.save

paddle.save(obj,
            path,
            protocol=4)

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射

PyTorch PaddlePaddle 备注
obj obj 要保存的对象实例，torch 支持 io.BytesIO、io.StringIO、文件，paddle 只支持文件，暂无转写方式。
f path 表示存储的路径。
pickle_module - 表示用于 pickling 元数据和对象的模块，Paddle 无此参数，暂无转写方式。
pickle_protocol protocol pickle 模块的协议版本。
_use_new_zipfile_serialization - 是否以旧格式加载文件，Paddle 无此参数，暂无转写方式。