diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py
index 3e36fb7632c41cfd61824fce11d6fb84c18ab1e1..4f73194d82e57336e1c496d9e5c6d40bb0eef15c 100644
--- a/python/paddle/fluid/layers/tensor.py
+++ b/python/paddle/fluid/layers/tensor.py
@@ -334,7 +334,7 @@ def assign(input, output=None, init_once=False):
             values = [int(v) for v in input.flat]
         else:
             raise ValueError("Unsupported dtype %s", input.dtype)
-        if input.size > 1024 * 1024:
+        if input.size > 1024 * 1024 * 5:
             raise ValueError("The size of input is too big. Please consider "
                              "saving it to file and 'load_op' to load it")