From a39541623a2bd7150d69a215c2e56e041dc00397 Mon Sep 17 00:00:00 2001
From: Chris McCarthy <chris.mccarthy@methods.co.uk>
Date: Fri, 21 Jul 2023 15:07:21 +0100
Subject: [PATCH] #1648 - Reverted benchmark sessions and episodes numbers

---
 benchmark/config/benchmark_training_config.yaml | 2 +-
 benchmark/primaite_benchmark.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/benchmark/config/benchmark_training_config.yaml b/benchmark/config/benchmark_training_config.yaml
index a56317f8..02b6377c 100644
--- a/benchmark/config/benchmark_training_config.yaml
+++ b/benchmark/config/benchmark_training_config.yaml
@@ -63,7 +63,7 @@ observation_space:
     - name: LINK_TRAFFIC_LEVELS
 
 # Number of episodes for training to run per session
-num_train_episodes: 10
+num_train_episodes: 500
 
 # Number of time_steps for training per episode
 num_train_steps: 256
diff --git a/benchmark/primaite_benchmark.py b/benchmark/primaite_benchmark.py
index c34add53..ead5723b 100644
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -414,7 +414,7 @@ def run():
     """Run the PrimAITE benchmark."""
     start_datetime = datetime.now()
     av_reward_per_episode_dicts = {}
-    for i in range(1, 3):
+    for i in range(1, 11):
         print(f"Starting Benchmark Session: {i}")
         with _get_benchmark_primaite_session() as session:
             session.learn()