pere's picture
eval updated
81de315
PROJECT_DIR=${HOME}"/models/t5-parliament-categorisation"
#EVAL_OUTPUT_DIR="gs://nb-t5x/eval/"
T5X_DIR="../../t5x" # directory where the t5x is cloned.
CHECKPOINT_PATH="gs://nb-t5x/eval_norwegian_NCC_2_000_000/checkpoint_2005000"
export PYTHONPATH=${PROJECT_DIR}
python3 eval.py \
--gin_search_paths=${PROJECT_DIR} \
--gin_file="eval_categorisation_base.gin" \
--gin.CHECKPOINT_PATH=\"${CHECKPOINT_PATH}\" \
# --gin.EVAL_OUTPUT_DIR=\"${EVAL_OUTPUT_DIR}\" \