昇腾社区首页
中文
注册
开发者
下载

(beta)torch_npu.utils.save_async

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]
[object Object]Atlas 推理系列产品[object Object]
[object Object]Atlas 训练系列产品[object Object]

功能说明

异步保存一个对象到一个硬盘文件上。

函数原型

[object Object]

参数说明

  • obj (object):要保存的对象。
  • f (Union[str, PathLike, BinaryIO, IO[bytes]]):保存的目标文件路径或文件句柄。
  • pickle_module (Any):用于Pickle序列化的模块,默认值为pickle。
  • pickle_protocol (int):Pickle协议版本,默认值为DEFAULT_PROTOCOL。
  • _use_new_zipfile_serialization (bool):是否使用新的Zip文件序列化,默认值为True。
  • _disable_byteorder_record (bool):是否禁用字节顺序记录,默认值为False。
  • model:模型对象,需要对模型反向注册hook,torch.nn.Module,默认值为None。

约束说明

该接口仅支持PyTorch 2.1.0版本。

调用示例

[object Object]