From 4249314672062c6edc8854e2ec9980075a3f6091 Mon Sep 17 00:00:00 2001 From: Chris McCarthy Date: Tue, 25 Jun 2024 17:28:05 +0100 Subject: [PATCH] #2648 - reduced the benchmark sessions and episodes for fail-fast speed while testing --- benchmark/primaite_benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmark/primaite_benchmark.py b/benchmark/primaite_benchmark.py index f3d0a10c..92c9bf0a 100644 --- 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,