#!/bin/bash

data_root='your_path'
testsets=A/V/R/K
csv_loc='./log/dem_seed0_rn_ds.csv'
arch=RN50
bs=64
ctx_init=a_photo_of_a
run_type=tpt_ours
lambda_term=1e5
seed=0
gpu=$1

python ./ours_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/dem_seed1_rn_ds.csv'
seed=1

python ./ours_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/dem_seed2_rn_ds.csv'
seed=2

python ./ours_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} \