python Mxnet Viz

Posted 2021-05-08

# Symbol
x = mx.sym.var('data')
y = net(x)
mx.viz.print_summary(y, shape={"data": (1, 3, 112, 112)})
graph=mx.viz.plot_network(y)
graph.render()

# SymbolBlock
x=mx.nd.random_normal(shape=(1,3,112,112))
net.summary(x)
#convert to Symbol
net.hybridize()
x = mx.sym.var('data')
y = net(x)
# y = mx.sym.Group(y)  # muti output
mx.viz.print_summary(y, shape={"data": (1, 3, 112, 112)})
graph=mx.viz.plot_network(y)
graph.render()