gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

adding a flag to prevent summary

Browse Source
This commit is contained in:
Itai Caspi
2018-02-14 12:30:25 +02:00
committed by Itai Caspi
parent 55c8c87afc
commit 72d34f4063
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
coach.py
View File

@@ -237,6 +237,9 @@ if __name__ == "__main__":
parser.add_argument('-tb', '--tensorboard', parser.add_argument('-tb', '--tensorboard',
help="(flag) When using the TensorFlow backend, enable TensorBoard log dumps. ", help="(flag) When using the TensorFlow backend, enable TensorBoard log dumps. ",
action='store_true') 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) args, run_dict = check_input_and_fill_run_dict(parser)
@@ -246,6 +249,7 @@ if __name__ == "__main__":
# dump documentation # dump documentation
logger.set_dump_dir(run_dict['experiment_path'], add_timestamp=True) logger.set_dump_dir(run_dict['experiment_path'], add_timestamp=True)
if not args.no_summary:
atexit.register(logger.print_summary) atexit.register(logger.print_summary)
# Single-threaded runs # Single-threaded runs