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

Merge pull request !19829 from Xiaoda/75-fix-the-host-device-problem-on-master
This commit is contained in:
i-robot 2021-07-10 06:40:57 +00:00 committed by Gitee
commit 8ab30e0638
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()