#!/bin/bash

#SBATCH --array=0-4%10
#SBATCH --time=1-0
#SBATCH --mem=64G
#SBATCH --gpus=1
#SBATCH --partition=RTX3090,RTXA6000,A100-PCI,A100-40GB,H100-PCI
#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"
#SBATCH --output=/SBATCH/OUT/FOLDER/%x-%j-%N-%a.out

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