python-2.7 - mx.viz.plot_network，无法绘制 resnet

Question

伙计们，我从这个网址下载 resnet-152：http: //data.mxnet.io/models/imagenet-11k/resnet-152/

然后，我尝试以这种方式绘制这个 152 层的 resnet：

import mxnet as mx 

sym, arg_params, aux_params = mx.model.load_checkpoint('resnet-152', 0)
pltn = mx.viz.plot_network(sym, shape = {'data': (1, 3, 224, 224)}, \
node_attrs = {'shape': 'oval', 'fixedsize': 'false'})
pltn.view('resnet-152')

但是，输出的 pdf 文件是空的。我可以绘制 LeNet、AlexNet 和 VGG，但是 resnet 有什么问题？非常感谢。

Answer 1

我可以确认所编写的代码适用于 MXNet 1.1.0，并输出包含整个神经网络图的 PDF。