在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')
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529