File size: 347 Bytes
680ab9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/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}" "$@" |