detailed error output for parameter sharing

Reviewed By: xianjiec Differential Revision: D6986239 fbshipit-source-id: 5b8bb06ea2383ce64318b5322bda7a58469f3eb0
walterddr · Feb 14, 2018 · 0a66c76 · 0a66c76
1 parent c784a27
commit 0a66c76
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/caffe2/python/layer_model_helper.py b/caffe2/python/layer_model_helper.py
@@ -213,8 +213,9 @@ def _validate_param_shape(self, param_name, shape):
         if shape != ref_shape:
             raise ValueError(
                 "Got inconsistent shapes between shared parameters "
-                "when trying to map a blob in scope {0} to {1}.".format(
-                    scope.CurrentNameScope(), param_name)
+                "when trying to map a blob in scope {0} to {1}. ref_shape : "
+                " {2}, shape : {3}".format(
+                    scope.CurrentNameScope(), param_name, ref_shape, shape)
             )
 
     def create_param(self, param_name, shape, initializer, optimizer=None,