1 files changed, 24 insertions, 0 deletions
diff --git a/scripts/test_optimized.sbatch b/scripts/test_optimized.sbatch
new file mode 100644
index 0000000..bc86f33
--- /dev/null
+++ b/scripts/test_optimized.sbatch
@@ -0,0 +1,24 @@
+#!/bin/bash
+#SBATCH --job-name=test_opt
+#SBATCH --account=bfqt-delta-gpu
+#SBATCH --partition=gpuA40x4
+#SBATCH --nodes=1
+#SBATCH --ntasks=1
+#SBATCH --cpus-per-task=4
+#SBATCH --gpus-per-node=1
+#SBATCH --mem=16G
+#SBATCH --time=00:10:00
+#SBATCH --output=runs/slurm_logs/%j_test_opt.out
+#SBATCH --error=runs/slurm_logs/%j_test_opt.err
+
+set -e
+
+PROJECT_DIR="/projects/bfqt/users/yurenh2/ml-projects/snn-training"
+cd "$PROJECT_DIR"
+
+mkdir -p runs/slurm_logs
+
+echo "Testing optimized SpikingVGG forward..."
+python scripts/test_optimized_forward.py
+
+echo "Done!"