Add parameter name to AssertionError for deferred shape inference (#1…

…8537)
apache · Jun 18, 2020 · 14aeb38 · 14aeb38
1 parent 9591436
commit 14aeb38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/gluon/parameter.py b/python/mxnet/gluon/parameter.py
@@ -194,8 +194,8 @@ def shape(self, new_shape):
 
         assert len(self._shape) == len(new_shape) and \
             all(j in (-1, 0, i) for i, j in zip(new_shape, self._shape)), \
-            "Expected shape %s is incompatible with given shape %s."%(
-                str(new_shape), str(self._shape))  # -1 means unknown dim size in np_shape mode
+            "Expected shape %s is incompatible with given shape %s for Parameter %s."%(
+                str(new_shape), str(self._shape), str(self.name))  # -1 means unknown dim size in np_shape mode
 
         self._shape = new_shape