File size: 977 Bytes
a2ffd07 | 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 | #!/usr/bin/env bash
# Build the original-model output cache from caption_targets.json.
# Run this once before running any eval script to avoid loading the base model.
#
# Usage:
# bash experiment/scripts/data/build_original_cache.sh
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
source "${SCRIPT_DIR}/../baselines/_common.sh"
CAPTION_TARGETS="${CAPTION_TARGETS:-${PROJECT_ROOT}/experiment/data/caption_targets.json}"
ORIGINAL_CACHE_DIR="${ORIGINAL_CACHE_DIR:-./cached_original_outputs}"
if [ ! -f "${CAPTION_TARGETS}" ]; then
echo "ERROR: caption_targets.json not found at ${CAPTION_TARGETS}"
echo " Run experiment/scripts/data/run_build_caption_targets.sh first."
exit 1
fi
echo "Building original cache..."
echo " caption_targets: ${CAPTION_TARGETS}"
echo " output_dir: ${ORIGINAL_CACHE_DIR}"
python -m experiment.data.build_original_cache \
--caption_targets "${CAPTION_TARGETS}" \
--output_dir "${ORIGINAL_CACHE_DIR}" \
--use_val_split
|