#!/bin/bash

data_root='./dataset'
testsets=$1 # I A V R S
# arch=ViT-B/16
arch=RN50
bs=64
ctx_init=a_photo_of_a
run_type=tpt

# TPT 
CUDA_VISIBLE_DEVICES=$2 python ./tpt_classification.py ${data_root} --test_sets ${testsets} \
-a ${arch} -b ${bs} --gpu 0 \
--tpt --ctx_init ${ctx_init} --run_type ${run_type} --I_augmix \

# TPT + REACT
CUDA_VISIBLE_DEVICES=$2 python ./tpt_classification_react.py ${data_root} --test_sets ${testsets} \
-a ${arch} -b ${bs} --gpu 0 \
--tpt --ctx_init ${ctx_init} --run_type ${run_type} --I_augmix --labeling_budget 0.1