#!/bin/bash
# example: sh scripts/test_tpt_CDG.sh DTD / Flower102

data_root='../TPT3.0/DATA'
testsets=$1
# arch=RN50
arch=ViT-B/16
# MP-TPT-S
bs=8
local_views=0.1
language_views=4
selection_p=1
visual_prior_factor=0.1
inference_factor=0.1

# MP-TPT-S
# bs=64
# local_views=0.3
# language_views=32
# selection_p=0.3
# visual_prior_factor=0.1
# inference_factor=0.1

ctx_init=a_photo_of_a

python ./mptpt.py ${data_root} --test_sets ${testsets} \
-a ${arch} -b ${bs} --gpu 1 \
--tpt --ctx_init ${ctx_init} \
--selection_p ${selection_p} \
--local_views ${local_views} \
--language_views ${language_views} \
--visual_prior_factor ${visual_prior_factor} \
--inference_factor ${inference_factor}