From 47d15a307b50d64f65b4941ee2f1cb2ee43cdaa0 Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Fri, 10 Sep 2021 16:23:44 +0800 Subject: [PATCH] set gradient_merge_cond persistable to false (#35578) --- python/paddle/fluid/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py index c0b99b3a27..bd5daf35a7 100755 --- a/python/paddle/fluid/optimizer.py +++ b/python/paddle/fluid/optimizer.py @@ -6918,7 +6918,7 @@ class GradientMergeOptimizer(object): shape=[1], value=bool(0), dtype='bool', - persistable=True, + persistable=False, force_cpu=True) with device_guard("cpu"): -- GitLab