(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]