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