megengine.load

load(f, map_location=None, pickle_module=pickle)[源代码]

从文件中加载用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 前完成。

例如: