From 1ab5c51122bf8d37b397847f7cbd11e9ffe5e9df Mon Sep 17 00:00:00 2001
From: ShenLiang <1422485404@qq.com>
Date: Mon, 27 Dec 2021 20:38:26 +0800
Subject: [PATCH] fix bugs in fp16 for dp (#38405)

---
 python/paddle/fluid/dygraph/parallel.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/dygraph/parallel.py b/python/paddle/fluid/dygraph/parallel.py
index 81bb812202c..809a4d385ff 100644
--- a/python/paddle/fluid/dygraph/parallel.py
+++ b/python/paddle/fluid/dygraph/parallel.py
@@ -356,7 +356,8 @@ def sync_params_buffers(model,
                         src_rank=0,
                         is_model_parallel=False):
     model_vars = []
-    for _, param in model.state_dict().items():
+    params_buffers = model.parameters() + model.buffers()
+    for param in params_buffers:
         if not isinstance(param, core.VarBase):
             raise TypeError("The data type of '%s' must be Varbase" %
                             param.name)
-- 
GitLab