File size: 696 Bytes
5c8ef86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#!/bin/bash
#SBATCH --job-name=mbench_gpt_ref-ytvos_numbered_final
#SBATCH --partition=a4000
#SBATCH --nodelist=node05
#SBATCH --gres=gpu:1
#SBATCH --time=14-00:00:00
#SBATCH --mem=5G
#SBATCH --cpus-per-task=4
#SBATCH --output=/home/yejin/data/projects/yejin/VerbCentric_RIS/ReferFormer/slurm_log/mbench_gpt_ref-ytvos_numbered_final.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 mbench/gpt_ref-ytvos_numbered_cy_sanity_2.py \
--save_caption_path mbench/numbered_captions_gpt-4o_final.json \
--save_valid_obj_ids_path mbench/numbered_valid_obj_ids_gpt-4o_final.json |