diff options
| author | Nico Galoppo <nico.galoppo@intel.com> | 2016-11-21 11:03:52 -0800 |
|---|---|---|
| committer | Nico Galoppo <nico.galoppo@intel.com> | 2016-11-21 11:03:52 -0800 |
| commit | db6cf0a728cad63c93b345f2203f3ad1f5d5c2f4 (patch) | |
| tree | f75c3be9eb9db047a943e9f79c64bfc24c56f3db | |
| parent | 473f143f9422e7fc66e9590da6b2a1bb88e50b2f (diff) | |
| download | caffeonacl-db6cf0a728cad63c93b345f2203f3ad1f5d5c2f4.tar.gz caffeonacl-db6cf0a728cad63c93b345f2203f3ad1f5d5c2f4.tar.bz2 caffeonacl-db6cf0a728cad63c93b345f2203f3ad1f5d5c2f4.zip | |
Fix Python net drawing script
| -rw-r--r-- | python/caffe/draw.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/caffe/draw.py b/python/caffe/draw.py index 9eecf6d7..e4fd7aac 100644 --- a/python/caffe/draw.py +++ b/python/caffe/draw.py @@ -104,11 +104,11 @@ def get_layer_label(layer, rankdir): pooling_types_dict[layer.pooling_param.pool], layer.type, separator, - layer.pooling_param.kernel_size, + layer.pooling_param.kernel_size[0] if len(layer.pooling_param.kernel_size._values) else 1, separator, - layer.pooling_param.stride, + layer.pooling_param.stride[0] if len(layer.pooling_param.stride._values) else 1, separator, - layer.pooling_param.pad) + layer.pooling_param.pad[0] if len(layer.pooling_param.pad._values) else 0) else: node_label = '"%s%s(%s)"' % (layer.name, separator, layer.type) return node_label |