MXNet导出可视化网络结构

导出可视化图

import sys
sys.path.append('/home/')
from models.SelfNestingNet import *
from my_utils import *

net = mx.sym.Variable('data')
net = SelfNestingNet(nClasses=10,basicBlockType ='BasicUnit' )
mx.viz.plot_network(net(mx.sym.var("img")))
digraph = mx.viz.plot_network(net, shape={"data":(1, 1, 32, 32)},node_attrs={"fixedsize":"false"})
digraph.view()

错误：

Symbol only support integer index to fetch i-th output

ndarray, x = nd.random.uniform(shape=(1,3,224,224),ctx=mx.gpu())

这个是MXNet根据动态图的一次forward计算, 生成静态图的流程问题. 可以理解成:

1.hybridize(), 表示程序希望生成高效的静态图模型;
2.net(mxnet.ndarray.xxx()), 在这一次forward中程序能计算其所需的资源, 并写入到静态图的信息中;
3.再export()