#!/bin/bash

srun -K \
  --partition=H200,H100,A100-40GB,A100-80GB \
  --job-name="Segment ImageNet" \
  --nodes=1 \
  --gpus=1 \
  --cpus-per-task=16 \
  --mem=64G \
  --time=1-0 \
  --export=HF_HOME=/PATH/TO/HF_HOME/,NLTK_DATA=/PATH/TO/NLTK_DATA/ \
  --container-image=/PATH/TO.sqsh \
  --container-workdir="$(pwd)" \
  --container-mounts=/SET/CONTAINER/MOUNTS,"$(pwd)":"$(pwd)" \
  python3 segment_imagenet.py "$@"