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}\" \ | |