slurm updated

SainsburyWellcomeCentre · Jun 28, 2024 · 978f001 · 978f001
1 parent f67b4c2
commit 978f001
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 135 deletions.
diff --git a/examples/agent_examples/TEM_README.md b/examples/agent_examples/TEM_README.md
diff --git a/examples/agent_examples/whittington_2020_run.py b/examples/agent_examples/whittington_2020_run.py
@@ -14,8 +14,8 @@
 from neuralplayground.experiments import Sargolini2006Data
 
 # Set the location for saving the results of the simulation
-simulation_id = "TEM_custom_sim"
-save_path = os.path.join(os.getcwd(), "results_sim")
+simulation_id = "TEM_big_50K"
+save_path = os.path.join(os.getcwd(), "results_big_50K")
 # save_path = os.path.join(os.getcwd(), "examples", "agent_examples", "trained_results")
 agent_class = Whittington2020
 env_class = BatchEnvironment
@@ -26,40 +26,40 @@
 
 # Set the x and y limits for the arena
 arena_x_limits = [
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
 ]
 arena_y_limits = [
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
-    [-4, 4],
-    [-5, 5],
-    [-6, 6],
-    [-5, 5],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
+    [-8, 8],
+    [-10, 10],
+    [-12, 12],
+    [-10, 10],
 ]
 
 # Set parameters for the environment that generates observations

diff --git a/examples/agent_examples/whittington_2020_slurm.sh b/examples/agent_examples/whittington_2020_slurm.sh
diff --git a/examples/agent_examples/whittington_slurm.sh b/examples/agent_examples/whittington_slurm.sh
@@ -1,18 +1,21 @@
 #!/bin/bash
 #
-#SBATCH --job-name=TEM_log_test
-#SBATCH --mem 50000 # memory pool for all cores
+#SBATCH --job-name=$NAME
+#SBATCH --mem=50000 # memory pool for all cores
 #SBATCH --time=72:00:00 # time
-#SBATCH -o TEM_logs/TEM_log_test.%N.%j.out # STDOUT
-#SBATCH -e TEM_logs/TEM_log-test.%N.%j.err # STDERR
+#SBATCH -o TEM_logs/$NAME.%N.%j.out # STDOUT
+#SBATCH -e TEM_logs/$NAME.%N.%j.err # STDERR
 #SBATCH -p gpu
 #SBATCH --gres=gpu:1
 #SBATCH --ntasks=1
 
+# Set the job name variable
+NAME="TEM_big_50K"
+
 source ~/.bashrc
 
 conda activate NPG-env
 
 python whittington_2020_run.py
 
-exit
+exit