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

provide a command line option which prints the tuning_parameters to stdout

This commit is contained in:
Zach Dwiel
2018-01-10 16:28:41 -05:00
parent 9b963c86d0
commit c7b11f1e9a
2 changed files with 31 additions and 11 deletions

View File

@@ -1,5 +1,5 @@
#
# Copyright (c) 2017 Intel Corporation
# Copyright (c) 2017 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@@ -273,6 +273,9 @@ if __name__ == "__main__":
"\"visualization.render=False; num_training_iterations=500; optimizer='rmsprop'\"",
default=None,
type=str)
parser.add_argument('--print_parameters',
help="(flag) Print tuning_parameters to stdout",
action='store_true')
args, run_dict = check_input_and_fill_run_dict(parser)
@@ -290,6 +293,9 @@ if __name__ == "__main__":
tuning_parameters = json_to_preset(json_run_dict_path)
tuning_parameters.sess = set_framework(args.framework)
if args.print_parameters:
print('tuning_parameters', tuning_parameters)
# Single-thread runs
tuning_parameters.task_index = 0
env_instance = create_environment(tuning_parameters)
@@ -352,5 +358,3 @@ if __name__ == "__main__":
# wait for all workers
[w.wait() for w in workers]
evaluation_worker.kill()