20 lines
613 B
Bash
20 lines
613 B
Bash
#!/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 "$@"
|