#2648 - reduced the benchmark sessions and episodes for fail-fast speed while testing

2024-06-25 17:28:05 +01:00
parent a7731c94e3
commit 4249314672
1 changed files with 2 additions and 2 deletions
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -151,8 +151,8 @@ def _prepare_session_directory():


 def run(
-    number_of_sessions: int = 5,
-    num_episodes: int = 1000,
+    number_of_sessions: int = 1,
+    num_episodes: int = 25,
    episode_len: int = 128,
    n_steps: int = 1280,
    batch_size: int = 32,