eval/retrieval_eval.sh

#!/usr/bin/env bash

set -x
set -e

DIR="$( cd "$( dirname "$0" )" && cd .. && pwd )"
echo "working directory: ${DIR}"

MODEL_NAME_OR_PATH="piccolo-base-zh"
OUTPUT_DIR='Retrieval'

mkdir -p "${OUTPUT_DIR}"

python -u retrieval_eval.py \
    --model-name-or-path "${MODEL_NAME_OR_PATH}" \
    --pool-type avg \
    --output-dir "${OUTPUT_DIR}" "$@"