From 21f59779eab091c765595febf24065baacb7fdac Mon Sep 17 00:00:00 2001
From: xujiaqi01 <173596896@qq.com>
Date: Thu, 2 Apr 2020 21:40:43 +0800
Subject: [PATCH] fix dump slot in strategy (#23398)

---
 .../fleet/parameter_server/pslib/optimizer_factory.py         | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/incubate/fleet/parameter_server/pslib/optimizer_factory.py b/python/paddle/fluid/incubate/fleet/parameter_server/pslib/optimizer_factory.py
index 94d6508cfe4..11d56e84913 100644
--- a/python/paddle/fluid/incubate/fleet/parameter_server/pslib/optimizer_factory.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/pslib/optimizer_factory.py
@@ -530,7 +530,9 @@ class DistributedAdam(DistributedOptimizerImplBase):
         opt_info["dump_fields_path"] = strategy.get("dump_fields_path", "")
         opt_info["dump_param"] = strategy.get("dump_param", [])
         if server._server.downpour_server_param.downpour_table_param[
-                0].accessor.accessor_class == "DownpourCtrAccessor":
+                0].accessor.accessor_class in [
+                    "DownpourCtrAccessor", "DownpourCtrDoubleAccessor"
+                ]:
             opt_info["dump_slot"] = True
         elif server._server.downpour_server_param.downpour_table_param[
                 0].accessor.accessor_class == "DownpourSparseValueAccessor":
-- 
GitLab