load接口

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

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

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

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

Any

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

父主题： API接口参考