Spaces:
Build error
Build error
File size: 443 Bytes
6c25ddb 44a9d55 6c25ddb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
#!/usr/bin/env bash
set -e
set -x
python train.py --model=gen --ckpt_name='gen-RAMS-what-head' \
--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_head.jsonlines \
--train_batch_size=2 \
--eval_batch_size=4 \
--learning_rate=3e-5 \
--accumulate_grad_batches=4 \
--num_train_epochs=3 \
--mark_trigger
|