File size: 466 Bytes
4f33d95
81de315
4f33d95
b3a728f
4f33d95
 
b3a728f
4f33d95
 
 
81de315
1
2
3
4
5
6
7
8
9
10
11
12
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}\" \