megengine.save

save(obj, f, pickle_module=<module 'pickle' from '/opt/hostedtoolcache/Python/3.7.10/x64/lib/python3.7/pickle.py'>, pickle_protocol=4)

把对象保存成磁盘文件。

参数

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

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

• pickle_module – 默认值： pickle

• pickle_protocol – 默认值： pickle.HIGHEST_PROTOCOL

序列化（Serialization） megengine.load