raise ValueError("\"infer_output_file\" command line parameter is "
"required in inference mode")
infer_config.update(copy.deepcopy(config_module["infer_params"]))
deco_print("Inference config:")
pprint.pprint(infer_config)
if args.benchmark:
deco_print("Adjusting config for benchmarking")
After Change
raise ValueError("\"infer_output_file\" command line parameter is "
"required in inference mode")
infer_config.update(copy.deepcopy(config_module["infer_params"]))
if hvd is None or hvd.rank() == 0:
deco_print("Inference config:")
pprint.pprint(infer_config)
if args.benchmark:
deco_print("Adjusting config for benchmarking")
train_config["print_samples_steps"] = None
train_config["print_loss_steps"] = 1
