From ff1f6209d36ab4b0bbbcf1cf761f1df2f22d6ca1 Mon Sep 17 00:00:00 2001 From: Sven Eberhardt Date: Wed, 6 Apr 2016 11:34:50 -0400 Subject: [PATCH] Fix initialization of deconvolution layer parameters from python net_spec interface. The deconvolution layer uses the convolution_param subgroup for its parameters, so hardcode that connection into param_name_dict(). --- python/caffe/net_spec.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/caffe/net_spec.py b/python/caffe/net_spec.py index 63de4cce4b2..87a480e0843 100644 --- a/python/caffe/net_spec.py +++ b/python/caffe/net_spec.py @@ -37,7 +37,10 @@ def param_name_dict(): # strip the final '_param' or 'Parameter' param_names = [s[:-len('_param')] for s in param_names] param_type_names = [s[:-len('Parameter')] for s in param_type_names] - return dict(zip(param_type_names, param_names)) + result = dict(zip(param_type_names, param_names)) + # convolution param is re-used in deconvolution layer + result['Deconvolution'] = 'convolution' + return result def to_proto(*tops):