Team Ai
Apppublic

OFA-Sys/OFA-Visual_Grounding

sourceHugging Faceupdated 4y agoView on Hugging Face
45likes
evaluate_caption.sh30 linesDownload Raw Back to caption
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