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 94d6508cfe40c83c5e512063e1483f0a2996534a..11d56e84913a6caec6f48bb2d1141a9b7ad78275 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":