File size: 1,395 Bytes
6fc683c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
#!/bin/bash
set -ex
gpu_rank=$1
quantize_size=$2
model_path=$3
eval_json=$4
ann_json=$5
flickr_entities_path=$6
dir_path=`dirname $model_path`
file_name=`basename $model_path`
if [[ $eval_json == *"val"* ]]; then
eval_name="val"
else
eval_name="test"
fi
template_post_name=$(echo "$eval_json" | sed 's/.*\.json//' )
eval_out_post_name=".eval.flickr.${eval_name}${template_post_name}"
master_port=$((RANDOM%1000+20000))
cat $eval_json | CUDA_VISIBLE_DEVICES=$gpu_rank python -m torch.distributed.launch --nproc_per_node=1 --master_port=$master_port evaluation/caption_obj_few_shot.py None \
--task generation_obj \
--path $model_path \
--model-overrides "{'visual_pretrained': '',
'dict_path':'data/dict.txt'}" \
--dict-path 'data/dict.txt' \
--required-batch-size-multiple 1 \
--remove-bpe=sentencepiece \
--max-len-b 500 \
--add-bos-token \
--beam 1 \
--buffer-size 10 \
--image-feature-length 64 \
--locate-special-token 1 \
--batch-size 1 \
--nbest 1 \
--no-repeat-ngram-size 3 \
--location-bin-size $quantize_size > $dir_path/${file_name}$eval_out_post_name
python evaluation/flickr_entities/flickr_entities_evaluate.py \
$dir_path/${file_name}$eval_out_post_name \
--annotation_file $ann_json \
--flickr_entities_path $flickr_entities_path \
--quantized_size $quantize_size
|