Bart-gen-arg / train_rams.sh
adherent's picture
Add application file
4bb803b
raw
history blame
435 Bytes
#!/usr/bin/env bash
set -e
set -x
python train.py --model=gen --ckpt_name='gen-RAMS-4-span' \
--dataset=RAMS \
--train_file=data/RAMS_1.0/data/train.jsonlines \
--val_file=data/RAMS_1.0/data/dev.jsonlines \
--test_file=data/RAMS_1.0/data/test.jsonlines \
--train_batch_size=2 \
--eval_batch_size=4 \
--learning_rate=3e-5 \
--accumulate_grad_batches=4 \
--num_train_epochs=3 \
--mark_trigger