#!/bin/bash
data_root='./dataset'
testsets=$1
# arch=ViT-B/16
arch=RN50
bs=64
ctx_init=a_photo_of
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} \

# 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} --labeling_budget 0.1\
