HPSv2 / configs /HPSv2.sh
tgxs002's picture
init
54199b6
name=$0
. configs/controller.sh
args=" \
--zeroshot-frequency 1 \
--report-to tensorboard \
--train-data $local_ranking_path/train.json $local_benchmark_path/annotations.json \
--val-data $local_ranking_path/test.json $local_benchmark_path/annotations.json \
--train-folder $local_ranking_path/train $local_benchmark_path \
--val-folder $local_ranking_path/test $local_benchmark_path \
--warmup 500 \
--lr 0.0000033 \
--wd 0.35 \
--workers 4 4 \
--batch-size 16 16 \
--pretrained laion2B-s32B-b79K \
--dataset-type HPD ranking \
--ignore-in-train 0 1 \
--ignore-in-val 1 0 \
--train-data-sample-ratio 1.0 0 \
--model ViT-H-14 \
--lock-text \
--lock-image \
--lock-text-unlocked-layers 11 \
--lock-image-unlocked-groups 20 \
--logs none \
--light-augmentation \
--exp-name $name \
--iterations 100 \
"
eval "$header$args$extra_args 2>&1 | tee -a $work_dir/exp_$now.txt"