load接口

从文件中加载save/mult_save接口持久化的对象。

torch_mindio.load(path, open_way='memfs', map_location=None)

参数	是否必选	说明	取值要求
path	必选	加载路径。	有效UnderFS路径。
open_way	可选	加载方式。 memfs：使用MindIO ACP的高性能MemFS保存数据。 fopen：调用C标准库中的文件操作函数保存数据，通常作为memfs方式的备份存在。默认值：memfs。	memfs fopen
map_location	可选	加载时需要映射到的设备。默认值：None。	None cpu

参数

是否必选

说明

取值要求

path

必选

加载路径。

有效UnderFS路径。

open_way

可选

加载方式。

默认值：memfs。

map_location

可选

加载时需要映射到的设备。默认值：None。

>>> # load from file
>>> torch_mindio.load('/mnt/dpc01/checkpoint/rank-0.pt')

Any

如同PyTorch的load接口一样，本接口内部也使用pickle模块，有被恶意构造的数据在unpickle期间攻击的风险。需要保证被加载的数据来源是安全存储的，仅可以load可信的数据。

父主题： Torch