megengine.load

load(f, map_location=None, pickle_module=<module 'pickle' from '/opt/hostedtoolcache/Python/3.7.10/x64/lib/python3.7/pickle.py'>)

从文件中加载用save()保存的对象。

参数

• f (text file object) – 文件名字符串或一个需要加载的文件对象。

• map_location (str, dict or a function specifying the map rules) – 默认值：None

注解

map_location 定义了设备的映射。用法请参考示例。

参数

pickle_module – 默认值： pickle

注解

如果你要调用 mge.set_default_device，请在调用 mge.load 前完成。

例如：

megengine.save 随机（Random）