From cd99d6b874d9d09b3bb87b8485cc787885af71f1 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Tue, 13 Jan 2026 23:49:05 -0600 Subject: init commit --- scripts/test_optimized.sbatch | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 scripts/test_optimized.sbatch (limited to 'scripts/test_optimized.sbatch') 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!" -- cgit v1.2.3