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。