#2648 - reverted temp changes to benchmark durations and branch names for testing purposes

2024-06-26 12:52:29 +01:00
parent 795b5a80fb
commit e7f979b78e
2 changed files with 8 additions and 9 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 = 1,
-    num_episodes: int = 25,
+    number_of_sessions: int = 5,
+    num_episodes: int = 1000,
    episode_len: int = 128,
    n_steps: int = 1280,
    batch_size: int = 32,