mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
#
|
|
# Copyright (c) 2019 Intel Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
"""
|
|
Definitions file:
|
|
|
|
It's main functionality are:
|
|
1) housing project constants and enums.
|
|
2) housing configuration parameters.
|
|
3) housing resource paths.
|
|
"""
|
|
|
|
|
|
class Definitions:
|
|
GROUP_NAME = "rl_coach"
|
|
PROCESS_NAME = "coach"
|
|
DASHBOARD_PROC = "firefox"
|
|
|
|
class Flags:
|
|
css = "checkpoint_save_secs"
|
|
crd = "checkpoint_restore_dir"
|
|
et = "environment_type"
|
|
ept = "exploration_policy_type"
|
|
cp = "custom_parameter"
|
|
seed = "seed"
|
|
dccp = "distributed_coach_config_path"
|
|
|
|
"""
|
|
Arguments that can be tested for python coach command
|
|
** None = Flag - no need for string or int
|
|
** {} = Add format for this parameter
|
|
"""
|
|
cmd_args = {
|
|
# '-l': None,
|
|
# '-e': '{}',
|
|
# '-r': None,
|
|
# '-n': '{' + enw + '}',
|
|
# '-c': None,
|
|
# '-ew': None,
|
|
'--play': None,
|
|
'--evaluate': None,
|
|
# '-v': None,
|
|
# '-tfv': None,
|
|
'--nocolor': None,
|
|
# '-s': '{' + css + '}',
|
|
# '-crd': '{' + crd + '}',
|
|
'-dg': None,
|
|
'-dm': None,
|
|
# '-et': '{' + et + '}',
|
|
# '-ept': '{' + ept + '}',
|
|
# '-lvl': '{level}',
|
|
# '-cp': '{' + cp + '}',
|
|
'--print_networks_summary': None,
|
|
'-tb': None,
|
|
'-ns': None,
|
|
'-d': None,
|
|
# '--seed': '{' + seed + '}',
|
|
'-onnx': None,
|
|
'-dc': None,
|
|
# '-dcp': '{' + dccp + '}',
|
|
'-asc': None,
|
|
'--dump_worker_logs': None,
|
|
}
|
|
|
|
class Presets:
|
|
# Preset list for testing the flags/ arguments of python coach command
|
|
args_test = [
|
|
"CartPole_A3C",
|
|
# "CartPole_NEC",
|
|
]
|
|
|
|
class Path:
|
|
experiments = "./experiments"
|
|
tensorboard = "tensorboard"
|
|
gifs = "gifs"
|
|
videos = "videos"
|
|
|
|
class Consts:
|
|
ASSERT_MSG = "Expected: {}, Actual: {}."
|
|
RESULTS_SORTED = "Results stored at:"
|
|
TOTAL_RUNTIME = "Total runtime:"
|
|
DISCARD_EXP = "Do you want to discard the experiment results"
|
|
REACHED_REQ_ASC = "Reached required success rate. Exiting."
|
|
INPUT_EMBEDDER = "Input Embedder:"
|
|
MIDDLEWARE = "Middleware:"
|
|
OUTPUT_HEAD = "Output Head:"
|
|
ONNX_WARNING = "Exporting ONNX graphs requires setting the " \
|
|
"--checkpoint_save_secs flag. The --export_onnx_graph" \
|
|
" will have no effect."
|
|
COLOR_PREFIX = "## agent: Starting evaluation phase"
|
|
TRAINING = "Training - "
|
|
PLAY_WARNING = "Both the --preset and the --play flags were set. " \
|
|
"These flags can not be used together. For human " \
|
|
"control, please use the --play flag together with " \
|
|
"the environment type flag (-et)"
|
|
|
|
class TimeOuts:
|
|
test_time_limit = 60 * 60
|
|
wait_for_files = 20
|