未验证 提交 21f59779 编写于 作者: X xujiaqi01 提交者: GitHub

fix dump slot in strategy (#23398)

上级 f2178376
...@@ -530,7 +530,9 @@ class DistributedAdam(DistributedOptimizerImplBase): ...@@ -530,7 +530,9 @@ class DistributedAdam(DistributedOptimizerImplBase):
opt_info["dump_fields_path"] = strategy.get("dump_fields_path", "") opt_info["dump_fields_path"] = strategy.get("dump_fields_path", "")
opt_info["dump_param"] = strategy.get("dump_param", []) opt_info["dump_param"] = strategy.get("dump_param", [])
if server._server.downpour_server_param.downpour_table_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 opt_info["dump_slot"] = True
elif server._server.downpour_server_param.downpour_table_param[ elif server._server.downpour_server_param.downpour_table_param[
0].accessor.accessor_class == "DownpourSparseValueAccessor": 0].accessor.accessor_class == "DownpourSparseValueAccessor":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册