summaryrefslogtreecommitdiff
path: root/scripts/run_scaled_grid_d4.sbatch
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/run_scaled_grid_d4.sbatch')
-rw-r--r--scripts/run_scaled_grid_d4.sbatch42
1 files changed, 42 insertions, 0 deletions
diff --git a/scripts/run_scaled_grid_d4.sbatch b/scripts/run_scaled_grid_d4.sbatch
new file mode 100644
index 0000000..7629804
--- /dev/null
+++ b/scripts/run_scaled_grid_d4.sbatch
@@ -0,0 +1,42 @@
+#!/bin/bash
+#SBATCH --job-name=scaled_d4
+#SBATCH --account=bfqt-delta-gpu
+#SBATCH --partition=gpuA40x4
+#SBATCH --nodes=1
+#SBATCH --ntasks=1
+#SBATCH --cpus-per-task=8
+#SBATCH --gpus-per-node=1
+#SBATCH --mem=64G
+#SBATCH --time=12:00:00
+#SBATCH --output=runs/slurm_logs/%j_scaled_grid_d4.out
+#SBATCH --error=runs/slurm_logs/%j_scaled_grid_d4.err
+
+set -e
+
+PROJECT_DIR="/projects/bfqt/users/yurenh2/ml-projects/snn-training"
+cd "$PROJECT_DIR"
+
+mkdir -p runs/slurm_logs runs/scaled_grid data
+
+echo "============================================================"
+echo "SCALED REGULARIZATION GRID SEARCH - DEPTH 4"
+echo "Job ID: $SLURM_JOB_ID | Node: $SLURM_NODELIST"
+echo "Start: $(date)"
+echo "============================================================"
+echo "Grid: λ_reg=[0.0005, 0.001, 0.002, 0.005] × reg_type=[mult_linear, mult_log]"
+echo "Total: 8 experiments"
+echo "============================================================"
+nvidia-smi --query-gpu=name,memory.total --format=csv,noheader
+echo "============================================================"
+
+python files/experiments/scaled_reg_grid_search.py \
+ --depth 4 \
+ --epochs 100 \
+ --batch_size 128 \
+ --lr 0.001 \
+ --data_dir ./data \
+ --out_dir ./runs/scaled_grid
+
+echo "============================================================"
+echo "Finished: $(date)"
+echo "============================================================"