chore: Update notebook_benchmark parameters for different environments

Adjust the parameters for the notebook_benchmark script to run benchmarks on different environments. This includes changing the gymid, training_episode_count, eval_episode_count, maximum_total_credentials, and plots_dir values for each environment.

Co-authored-by: William Blum <[email protected]>

notebooks/run_quick.sh

@@ -44,31 +44,36 @@ run chainnetwork-random '' -y "
 "
 run randomnetwork ''
 
-run notebook_benchmark '-toyctf' -y "
+run notebook_benchmark '-chain' -y "
+    gymid: 'CyberBattleChain-v0'
     iteration_count: 100
-    training_episode_count: 3
-    eval_episode_count: 5
+    training_episode_count: 5
+    eval_episode_count: 3
     maximum_node_count: 12
-    maximum_total_credentials: 10
+    maximum_total_credentials: 7
     plots_dir: $output_plot_dir
 "
 
-run notebook_benchmark '-chain' -y "
+run notebook_benchmark '-toyctf' -y "
+    gymid: 'CyberBattleToyCtf-v0'
+    env_size: null
     iteration_count: 100
-    training_episode_count: 5
-    eval_episode_count: 3
+    training_episode_count: 3
+    eval_episode_count: 5
     maximum_node_count: 12
-    maximum_total_credentials: 7
+    maximum_total_credentials: 10
     plots_dir: $output_plot_dir
 "
 
 run notebook_benchmark '-tiny' -y "
-  iteration_count: 100
-  training_episode_count: 4
-  eval_episode_count: 2
-  maximum_node_count: 5
-  maximum_total_credentials: 3
-  plots_dir: $output_plot_dir
+    gymid: 'CyberBattleTiny-v0'
+    env_size: null
+    iteration_count: 100
+    training_episode_count: 4
+    eval_episode_count: 2
+    maximum_node_count: 5
+    maximum_total_credentials: 3
+    plots_dir: $output_plot_dir
 "
 
 run notebook_dql_transfer '' -y "