python-2.7 - mx.viz.plot_network、resnet をプロットできません

Question

皆さん、私はこの URL から resnet-152 をダウンロードします: http://data.mxnet.io/models/imagenet-11k/resnet-152/

次に、この resnet を 152 レイヤーで次のようにプロットしてみます。

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 を出力することを確認できます。