multi_save接口
接口功能
将同一个数据保存到多个文件中。
接口格式
mindio_acp.multi_save(obj, path_list)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
---|---|---|---|
obj |
必选 |
需要保存的对象。 |
有效数据对象。 |
path_list |
必选 |
数据保存路径列表。 |
有效文件路径列表。 |
使用样例
>>> # Save to file >>> x = torch.tensor([0, 1, 2, 3, 4]) >>> path_list = ["/mnt/dpc01/dir1/rank_1.pt","/mnt/dpc01/dir2/rank_1.pt"] >>> mindio_acp.multi_save(x, path_list)
返回值
- None:失败。
- 0:使用原生torch.save方式保存成功。
- 1:使用memfs方式保存成功。
- 2:使用fopen保存成功。
父主题: API接口参考