train_five_datasets_l2p.sh

#!/bin/bash

#SBATCH --job-name=l2p
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH -p batch
#SBATCH -w agi1
#SBATCH --cpus-per-gpu=4
#SBATCH --mem-per-gpu=20G
#SBATCH --time=14-0
#SBATCH -o %N_%x_%j.out
#SBTACH -e %N_%x_%j.err

source /data/jaeho/init.sh
conda activate torch38gpu
python -m torch.distributed.launch \
        --nproc_per_node=1 \
        --use_env main.py \
        five_datasets_l2p \
        --model vit_base_patch16_224 \
        --batch-size 16 \
        --data-path /local_datasets/ \
        --output_dir ./output