!19818 [Wide&Deep] Add the sparse configuration from the host_device flag

Merge pull request !19818 from Xiaoda/74-fix-the-host-device-problem
This commit is contained in:
i-robot 2021-07-10 03:40:56 +00:00 committed by Gitee
commit bfce9e226a
1 changed files with 4 additions and 1 deletions

View File

@ -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()