| #SBATCH --job-name=trial | |
| #SBATCH --gres=gpu:1 | |
| #SBATCH --partition=a6000 | |
| #SBATCH --time=0-12:00:00 # d-hh:mm:ss, job time limit | |
| #SBATCH --mem=20000 # cpu memory size | |
| #SBATCH --cpus-per-task=4 | |
| #SBATCH --output=./trial.log | |
| ml purge | |
| ml load cuda/11.8 | |
| eval "$(conda shell.bash hook)" | |
| conda activate risall | |
| cd /data2/projects/chaeyun/RIS-DMMI | |
| CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node 1 train.py --model dmmi_swin --dataset refcocog --splitBy umd --model_id 1234 --batch-size 2 --lr 0.00005 --wd 1e-2 --window12 --swin_type base --pretrained_backbone /data2/projects/chaeyun/LAVT-RIS/pretrained_weights/swin_base_patch4_window12_384_22k.pth --output_dir './tmp/' --epochs 40 --img_size 480 2>&1 |tee ./tmp/output.log |