#!/bin/bash
#SBATCH --job-name=train_custom
#SBATCH --output=logs/train_custom_%A_%a.out
#SBATCH --error=logs/train_custom_%A_%a.err
#SBATCH --array=0-15
#SBATCH --time=01:00:00
#SBATCH --cpus-per-task=1
#SBATCH --constraint="hsw"

TASK_ID=$SLURM_ARRAY_TASK_ID
DIM=$((10 + TASK_ID))

echo "Running task $TASK_ID with DIM=$DIM"

python -m scripts.tools.train \
    --dataset custom \
    --custom-dim "$DIM"
