update

Author: JiayiFeng

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