adjust the sparse config from the host_device_mix flag

2021-07-09 14:52:33 +08:00 · 2021-07-09 14:52:33 +08:00 · ddfcff6de2
parent 1f1065eab0
commit ddfcff6de2
1 changed files with 4 additions and 1 deletions
--- a/model_zoo/official/recommend/wide_and_deep/src/model_utils/config.py
+++ b/model_zoo/official/recommend/wide_and_deep/src/model_utils/config.py
@ -122,6 +122,9 @@ def get_config():
    pprint(default)
    args = parse_cli_to_yaml(parser=parser, cfg=default, helper=helper, choices=choices, cfg_path=path_args.config_path)
    final_config = merge(args, default)
-    return Config(final_config)
+    final_config = Config(final_config)
+    if final_config.host_device_mix == 1:
+        final_config.sparse = True
+    return final_config

 config = get_config()