Skip to content

Commit

Permalink
Merge pull request BVLC#3977 from tpwrules/master
Browse files Browse the repository at this point in the history
Fix protobuf message generation
  • Loading branch information
longjon committed May 4, 2016
2 parents 2f7fc59 + 09130ce commit 6f5f7c5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/caffe/net_spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def param_name_dict():
# get all parameter names (typically underscore case) and corresponding
# type names (typically camel case), which contain the layer names
# (note that not all parameters correspond to layers, but we'll ignore that)
param_names = [s for s in dir(layer) if s.endswith('_param')]
param_names = [f.name for f in layer.DESCRIPTOR.fields if f.name.endswith('_param')]
param_type_names = [type(getattr(layer, s)).__name__ for s in param_names]
# strip the final '_param' or 'Parameter'
param_names = [s[:-len('_param')] for s in param_names]
Expand Down

0 comments on commit 6f5f7c5

Please sign in to comment.