fix dump slot in strategy (#23398)

21f59779 · xujiaqi01 · GitHub · f2178376 · 21f59779
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

python/paddle/fluid/incubate/fleet/parameter_server/pslib/optimizer_factory.py ...ncubate/fleet/parameter_server/pslib/optimizer_factory.py +3 -1

未找到文件。
--- 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":