OFA-Sys/OFA-Visual_Grounding
45
1#!/usr/bin/env bash2 3user_dir=../../ofa_module4bpe_dir=../../utils/BPE5 6data=../../dataset/caption_data/caption_test.tsv7path=../../checkpoints/caption_large_best_clean.pt8result_path=../../results/caption9selected_cols=1,4,210split='test'11 12CUDA_VISIBLE_DEVICES=4,5,6,7 python3 ../../evaluate.py \13 ${data} \14 --path=${path} \15 --user-dir=${user_dir} \16 --task=caption \17 --batch-size=16 \18 --log-format=simple --log-interval=10 \19 --seed=7 \20 --gen-subset=${split} \21 --results-path=${result_path} \22 --beam=5 \23 --max-len-b=16 \24 --no-repeat-ngram-size=3 \25 --fp16 \26 --num-workers=0 \27 --model-overrides="{\"data\":\"${data}\",\"bpe_dir\":\"${bpe_dir}\",\"eval_cider\":False,\"selected_cols\":\"${selected_cols}\"}"28 29python coco_eval.py ../../results/caption/test_predict.json ../../dataset/caption_data/test_caption_coco_format.json30 