forked from mindspore-Ecosystem/mindspore
!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:
commit
8ab30e0638
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue