From 4371ca13fc0b655aeb2d4d20b365ef400f5db68b Mon Sep 17 00:00:00 2001 From: SunilSamra Date: Thu, 6 Jul 2023 11:12:51 +0100 Subject: [PATCH] 1566 - added train_episodes, train_steps, eval_episodes and eval_steps to training_config_main.yaml --- .../training/training_config_main.yaml | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/primaite/config/_package_data/training/training_config_main.yaml b/src/primaite/config/_package_data/training/training_config_main.yaml index a638fe14..dc94e3bb 100644 --- a/src/primaite/config/_package_data/training/training_config_main.yaml +++ b/src/primaite/config/_package_data/training/training_config_main.yaml @@ -49,11 +49,19 @@ observation_space: - name: NODE_LINK_TABLE # - name: NODE_STATUSES # - name: LINK_TRAFFIC_LEVELS -# Number of episodes to run per session -num_episodes: 10 -# Number of time_steps per episode -num_steps: 256 + +# Number of episodes for training to run per session +num_train_episodes: 10 + +# Number of time_steps for training per episode +num_train_steps: 256 + +# Number of episodes for evaluation to run per session +num_eval_episodes: 10 + +# Number of time_steps for evaluation per episode +num_eval_steps: 256 # Sets how often the agent will save a checkpoint (every n time episodes). # Set to 0 if no checkpoints are required. Default is 10