#1648 - Reverted benchmark sessions and episodes numbers

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

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()