load接口

接口功能

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

接口格式

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

接口参数

使用样例

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

返回值

Any

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