1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

adding a flag to prevent summary

This commit is contained in:
Itai Caspi
2018-02-14 12:30:25 +02:00
committed by Itai Caspi
parent 55c8c87afc
commit 72d34f4063

View File

@@ -237,6 +237,9 @@ if __name__ == "__main__":
parser.add_argument('-tb', '--tensorboard',
help="(flag) When using the TensorFlow backend, enable TensorBoard log dumps. ",
action='store_true')
parser.add_argument('-ns', '--no_summary',
help="(flag) Prevent Coach from printing a summary and asking questions at the end of runs",
action='store_true')
args, run_dict = check_input_and_fill_run_dict(parser)
@@ -246,6 +249,7 @@ if __name__ == "__main__":
# dump documentation
logger.set_dump_dir(run_dict['experiment_path'], add_timestamp=True)
if not args.no_summary:
atexit.register(logger.print_summary)
# Single-threaded runs