#!/bin/bash
# example: sh scripts/test_tpt_CDG.sh R

data_root='../TPT3.0/DATA'
testsets=$1
# arch=RN50
arch=ViT-B/16

bs=64
local_views=0.3
language_views=32
selection_p=0.1
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 0 \
--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}