xx-symbol.json # 模型结构的json文件
方法一:将json文件导入,将网络结构保存为pdf
import mxnet as mx
symbol = mx.symbol.load(xx-symbol.json)
mx.viz.plot_network(symbol,title='model',save_format='pdf',shape={'data':(1,3,224,224)}).view()
# shape参数需要去看一下json文件,data的维度为NCHW
xx-symbol.json # 模型结构的json文件
方法一:将json文件导入,将网络结构保存为pdf
import mxnet as mx
symbol = mx.symbol.load(xx-symbol.json)
mx.viz.plot_network(symbol,title='model',save_format='pdf',shape={'data':(1,3,224,224)}).view()
# shape参数需要去看一下json文件,data的维度为NCHW