From cd99d6b874d9d09b3bb87b8485cc787885af71f1 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Tue, 13 Jan 2026 23:49:05 -0600 Subject: init commit --- scripts/run_scaled_grid_d12.sbatch | 42 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 scripts/run_scaled_grid_d12.sbatch (limited to 'scripts/run_scaled_grid_d12.sbatch') diff --git a/scripts/run_scaled_grid_d12.sbatch b/scripts/run_scaled_grid_d12.sbatch new file mode 100644 index 0000000..eff5342 --- /dev/null +++ b/scripts/run_scaled_grid_d12.sbatch @@ -0,0 +1,42 @@ +#!/bin/bash +#SBATCH --job-name=scaled_d12 +#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=44:00:00 +#SBATCH --output=runs/slurm_logs/%j_scaled_grid_d12.out +#SBATCH --error=runs/slurm_logs/%j_scaled_grid_d12.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 12" +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 12 \ + --epochs 100 \ + --batch_size 128 \ + --lr 0.001 \ + --data_dir ./data \ + --out_dir ./runs/scaled_grid + +echo "============================================================" +echo "Finished: $(date)" +echo "============================================================" -- cgit v1.2.3