MXNet中怎么保存和加载模型
lewis
2020-01-13
33次阅读
在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')

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。