megengine.save

save(obj, f, pickle_module=pickle, pickle_protocol=pickle.HIGHEST_PROTOCOL)[源代码]

把对象保存成磁盘文件。

参数

  • obj – 要保存的对象。只支持 module 和 state_dict 。

  • f – 文件名字符串或一个需要保存的文件对象。

  • pickle_module – 默认值： pickle

  • pickle_protocol – 默认值： pickle.HIGHEST_PROTOCOL

上一页

megengine.core.autodiff.grad.get_grad_managers

下一页

megengine.load