mxnet(gluon)学习之路-网络可视化

如何可视化网络

神经网络的可视化是十分有用的，对于之前学习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()

结果如下所示：