#!/bin/bash

data_root='your_path'
testsets=A/V/R/K
csv_loc='./log/otpt_seed0_vit_ds.csv'
arch=ViT-B/16
bs=64
ctx_init=a_photo_of_a
run_type=tpt_otpt
lambda_term=2
gpu=$1
seed=0

python ./otpt_classification.py ${data_root} --test_sets ${testsets} --csv_log ${csv_loc} \
-a ${arch} -b ${bs} --gpu ${gpu} --seed ${seed} \
--tpt --ctx_init ${ctx_init} --run_type ${run_type} --I_augmix --lambda_term ${lambda_term} \

csv_loc='./log/otpt_seed1_vit_ds.csv'
seed=1

python ./otpt_classification.py ${data_root} --test_sets ${testsets} --csv_log ${csv_loc} \
-a ${arch} -b ${bs} --gpu ${gpu} --seed ${seed} \
--tpt --ctx_init ${ctx_init} --run_type ${run_type} --I_augmix --lambda_term ${lambda_term} \

csv_loc='./log/otpt_seed2_vit_ds.csv'
seed=2

python ./otpt_classification.py ${data_root} --test_sets ${testsets} --csv_log ${csv_loc} \
-a ${arch} -b ${bs} --gpu ${gpu} --seed ${seed} \
--tpt --ctx_init ${ctx_init} --run_type ${run_type} --I_augmix --lambda_term ${lambda_term} \