#!/bin/bash

#SBATCH --array=0-29%30
#SBATCH --time=1-0
#SBATCH --mem=64G
#SBATCH --gpus=1
#SBATCH --partition=H200,H100,H100-PCI,A100-40GB,A100-80GB
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=16
#SBATCH --export="HF_HOME=/PATH/TO/HF_HOME/,NLTK_DATA=/PATH/TO/NLTK_DATA/,TQDM_DISABLE=1"
#SBATCH --job-name="Segment ImageNet (val)"
#SBATCH --output=/SBATCH/OUT/FOLDER/%x-%j-%N-%a.out
#SBATCH --wait

srun -K \
  --container-image=/PATH/TO.sqsh \
  --container-workdir="$(pwd)" \
  --container-mounts=/SET/CONTAINER/MOUNTS,"$(pwd)":"$(pwd)" \
  python3 segment_imagenet.py -p 30 -id $SLURM_ARRAY_TASK_ID -o /OUTPUT/FOLDER/PATH/ "$@"