（beta）torch_npu.utils.save_async

函数原型

功能说明

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

参数说明

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

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]

约束说明

该接口仅支持PyTorch 2.1.0版本。

调用示例