在MXNet中,可以使用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')
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: MXNet处理图像数据的方法是什么