AAAI Version

AAAI Supplementary Material/ForNet Creation Code/sbatch-segment-tinyimnet

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+#SBATCH --array=0-19%20
+#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 TinyImageNet"
+#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 20 -id $SLURM_ARRAY_TASK_ID "$@"