Spaces:
Build error
Build error
File size: 837 Bytes
6c25ddb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
#!/usr/bin/env bash
set -e
set -x
CKPT_NAME=gen-KAIROS
MODEL=constrained-gen
rm -rf checkpoints/${CKPT_NAME}-pred
python train.py --model=$MODEL --ckpt_name=${CKPT_NAME}-pred \
--load_ckpt=checkpoints/${CKPT_NAME}/epoch=2-v0.ckpt \
--dataset=KAIROS \
--eval_only \
--mark_trigger \
--train_file=data/wikievents/train.jsonl \
--val_file=data/wikievents/dev.jsonl \
--test_file=data/wikievents/test.jsonl \
--coref_dir=data/wikievents/coref \
--train_batch_size=4 \
--eval_batch_size=4 \
--learning_rate=3e-5 \
--accumulate_grad_batches=4 \
--num_train_epochs=3
python src/genie/scorer.py --gen-file=checkpoints/$CKPT_NAME-pred/predictions.jsonl \
--test-file=data/wikievents/test.jsonl \
--dataset=KAIROS \
--coref-file=data/wikievents/coref/test.jsonlines \
--head-only \
--coref
|