diff --git a/python/paddle/fluid/contrib/memory_usage_calc.py b/python/paddle/fluid/contrib/memory_usage_calc.py
index 5ffdca82bde837d1bf5627b9e17f1eccabdb94ae..baa14a573fcfdfa943af1e995f687c74e9fb4d07 100644
--- a/python/paddle/fluid/contrib/memory_usage_calc.py
+++ b/python/paddle/fluid/contrib/memory_usage_calc.py
@@ -87,8 +87,13 @@ def memory_usage(program, batch_size):
                 continue
 
             data_count = 1
+            neg_dim_count = 0
             for x in var.shape:
                 if x < 0:
+                    if neg_dim_count >= 1:
+                        raise ValueError("Var %s has more than one negtive dim."
+                                         % (var_name))
+                    neg_dim_count += 1
                     data_count *= batch_size * (-x)
                 else:
                     data_count *= x