伙计们,我从这个网址下载了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有什么问题?非常感谢你。
答案 0 :(得分:1)
我可以确认所写的代码适用于MXNet 1.1.0,并输出包含整个神经网络图形的PDF。