mxnet加载模型参数方法以及保存方法总结

采用model方式

加载模型和参数

mxnet加载模型参数方法以及保存方法总结_第1张图片

mxnet.module.Module.load

mxnet加载模型参数方法以及保存方法总结_第2张图片

保存模型和参数

mxnet加载模型参数方法以及保存方法总结_第3张图片

mxnet.module.Module.save_checkpoint

mxnet加载模型参数方法以及保存方法总结_第4张图片

采用gluon.SymbolBlock方式

  • 加载模型和参数

mxnet.gluon.SymbolBlock.imports

mxnet加载模型参数方法以及保存方法总结_第5张图片

  • 保存模型和参数

mxnet.gluon.SymbolBlock.export

mxnet加载模型参数方法以及保存方法总结_第6张图片

 

  • 加载参数

mxnet.gluon.SymbolBlock.load_parameters

mxnet加载模型参数方法以及保存方法总结_第7张图片

  • 保存参数

mxnet.gluon.SymbolBlock.save_parameters

mxnet加载模型参数方法以及保存方法总结_第8张图片

采用symbol方式

  • 加载模型
mxnet加载模型参数方法以及保存方法总结_第9张图片 mxnet加载模型参数方法以及保存方法总结_第10张图片
  • 保存模型

mxnet.symbol.Symbol.save

mxnet加载模型参数方法以及保存方法总结_第11张图片

mxnet.symbol.Symbol.tojson

mxnet加载模型参数方法以及保存方法总结_第12张图片

 

你可能感兴趣的:(深度学习相关博文,mxnet,加载模型,加载参数)