#!/bin/bash

data_root='./dataset'
testsets=$1
arch=RN50
# arch=ViT-B/16
bs=64

ctx_init=a_photo_of
run_type=tpt_ctpt
lambda_term=50


# C-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} --lambda_term ${lambda_term} \

# C-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} --lambda_term ${lambda_term}  --labeling_budget 0.1
