在MXNet中,可以使用mx.model.save()函数来保存模型,使用mx.model.load()函数来加载模型。
mx.model.save()
mx.model.load()
保存模型示例:
import mxnet as mx # 保存模型 model.save('model_checkpoint')
加载模型示例:
import mxnet as mx # 加载模型 model_loaded = mx.model.load('model_checkpoint')