VRIS_vip / slurm_script /annotate_ref_ytvos.sh
dianecy's picture
Add files using upload-large-folder tool
5c8ef86 verified
#!/bin/bash
#SBATCH --job-name=annotate_ref_ytvos
#SBATCH --partition=a4000
#SBATCH --nodelist=node05
#SBATCH --gres=gpu:1
#SBATCH --time=0-06:00:00
#SBATCH --mem=5G
#SBATCH --cpus-per-task=4
#SBATCH --output=/home/yejin/data/projects/yejin/VerbCentric_RIS/ReferFormer/slurm_log/annotate_ref_ytvos2.out
cd /home/yejin/data/projects/yejin/VerbCentric_RIS/ReferFormer
ml purge
ml load cuda/12.1
eval "$(conda shell.bash hook)"
conda activate referformer
python3 annotate_ref_ytvos.py