Undo bug with nested_update
@@ -171,14 +171,12 @@ def main():
if args.mode == 'train' or args.mode == 'train_eval':
if 'train_params' in config_module:
train_config.update(copy.deepcopy(config_module['train_params']))
- nested_update(train_config, nest_dict(vars(config_update)))
if hvd is None or hvd.rank() == 0:
deco_print("Training config:")
pprint.pprint(train_config)
if args.mode == 'eval' or args.mode == 'train_eval':
if 'eval_params' in config_module:
eval_config.update(copy.deepcopy(config_module['eval_params']))
- nested_update(eval_config, nest_dict(vars(config_update)))
if hvd is None or hvd.rank() == 0:
deco_print("Evaluation config:")
pprint.pprint(eval_config)
@@ -187,7 +185,7 @@ def main():
raise ValueError("\"infer_output_file\" command line parameter is "
"required in inference mode")
infer_config.update(copy.deepcopy(config_module['infer_params']))
- nested_update(infer_config, nest_dict(vars(config_update)))
+
if hvd is None or hvd.rank() == 0:
deco_print("Inference config:")
pprint.pprint(infer_config)
