From 908684a535a162f54f3e01b449779dda1853de85 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Wed, 9 Jan 2019 14:27:49 +0800 Subject: [PATCH] change the largest size of assign --- python/paddle/fluid/layers/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py index 3e36fb7632..4f73194d82 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") -- GitLab