14 lines
362 B
Bash
14 lines
362 B
Bash
#!/bin/bash
|
|
|
|
srun -K \
|
|
--partition=RTX3090,A100-40GB,H100 \
|
|
--job-name="Segment ImageNet" \
|
|
--nodes=1 \
|
|
--mem=64G \
|
|
--time=1-0 \
|
|
--export=HF_HOME=/PATH/TO/HF_HOME/,NLTK_DATA=/PATH/TO/NLTK_DATA/,MAX_WORKERS=16 \
|
|
--container-image=/PATH/TO.sqsh \
|
|
--container-workdir="$(pwd)" \
|
|
--container-mounts=/SET/CONTAINER/MOUNTS,"$(pwd)":"$(pwd)" \
|
|
"$@"
|