diff options
| author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-10-16 17:59:13 -0700 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-10-16 17:59:13 -0700 |
| commit | 3363e2528706bde6dc469f42eb4edaf9f02d97da (patch) | |
| tree | 25858681c7ec3840a0b17c704e37dc17f90ee978 | |
| parent | f9b2f1cdcbb41846e93d77f50fe65b7adde9c79a (diff) | |
| parent | e0c34cedde6e0d12d420a51cea7a98df50069559 (diff) | |
| download | caffeonacl-3363e2528706bde6dc469f42eb4edaf9f02d97da.tar.gz caffeonacl-3363e2528706bde6dc469f42eb4edaf9f02d97da.tar.bz2 caffeonacl-3363e2528706bde6dc469f42eb4edaf9f02d97da.zip | |
Merge pull request #3190 from DrSleep/master
Fix drawing problems with repeated convolution -- close #3201
| -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 a002b60b..f8bf5722 100644 --- a/python/caffe/draw.py +++ b/python/caffe/draw.py @@ -82,11 +82,11 @@ def get_layer_label(layer, rankdir): separator, layer.type, separator, - layer.convolution_param.kernel_size, + layer.convolution_param.kernel_size[0] if len(layer.convolution_param.kernel_size._values) else 1, separator, - layer.convolution_param.stride, + layer.convolution_param.stride[0] if len(layer.convolution_param.stride._values) else 1, separator, - layer.convolution_param.pad) + layer.convolution_param.pad[0] if len(layer.convolution_param.pad._values) else 0) elif layer.type == 'Pooling': pooling_types_dict = get_pooling_types_dict() node_label = '"%s%s(%s %s)%skernel size: %d%sstride: %d%spad: %d"' %\ |