VRIS_vip / slurm_script /mbench_gtp_ref-ytvos_numbered.sh
dianecy's picture
Add files using upload-large-folder tool
5c8ef86 verified
#!/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