File size: 517 Bytes
3d5551b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/bin/bash
# Path to trained model
MODEL_PATH="logs/model_glen_vault/GLEN_P2_base"
CUDA_VISIBLE_DEVICES=0 \
python examples/glen_phase1/evaluate_glen.py \
--model_name_or_path ${MODEL_PATH} \
--dataset_name the_vault \
--test100 0 \
--id_class t5_bm25_truncate_3 \
--max_input_length 32 \
--per_device_eval_batch_size 32 \
--output_dir ${MODEL_PATH} \
--tie_word_embeddings True \
--gen_method greedy \
--num_return_sequences 100 \
--tree 1 \
--overwrite_output_dir |