train.sh

#!/bin/bash

# SBATCH --job-name=dualprompt
# SBATCH --nodes=1
# SBATCH --gres=gpu:2
# SBATCH -p batch
# SBATCH -w agi1
# SBATCH --cpus-per-gpu=4
# SBATCH --mem-per-gpu=20G
# SBATCH --time=8:00:00
# SBATCH -o %x_%j.out
# SBTACH -e %x_%j.err

# source /data/jaeho/init.sh
# conda activate torch38gpu
python -m torch.distributed.launch \
        --nproc_per_node=2 \
        --use_env main.py \
#         --model vit_base_patch16_224 \
#         --batch-size 24 \
#         --data-path /local_datasets/ \
#         --output_dir ./output \
#         --epochs 5