diff --git a/official/nlp/transformer/src/model_utils/config.py b/official/nlp/transformer/src/model_utils/config.py index e3a8262e7473c0c6da7af1d931d986d296f852c9..5dff477aed4c7ea9af926186b82d84505024c2fe 100644 --- a/official/nlp/transformer/src/model_utils/config.py +++ b/official/nlp/transformer/src/model_utils/config.py @@ -122,7 +122,7 @@ def get_config(): args = parse_cli_to_yaml(parser=parser, cfg=default, helper=helper, choices=choices, cfg_path=path_args.config_path) final_config = merge(args, default) pprint(final_config) - print("Please check the above information for the configurations", flush=True) + pprint("Please check the above information for the configurations") return Config(final_config) config = get_config()