如何可视化网络
神经网络的可视化是十分有用的,对于之前学习caffe的同学来说可以使用 netscope来进行可视化,而对于mxnet来说我们可以使用mxnet自身的函数完成可视化。
import mxnet as mx
from mxnet import gluon
num_hidden = 64
net = gluon.nn.HybridSequential()
with net.name_scope():
net.add(gluon.nn.Dense(num_hidden, activation="relu"))
net.add(gluon.nn.Dense(num_hidden, activation="relu"))
net.add(gluon.nn.Dense(10))
net.hybridize()
net.collect_params().initialize()
x = mx.sym.var('data')
sym = net(x)
mx.viz.plot_network(sym).view() #pycharm使用必须加view()
结果如下所示: