Spaces:

adherent
/

Bart-gen-arg

Build error

App Files Files Community

Bart-gen-arg / train_rams.sh

adherent's picture

Add application file

4bb803b over 1 year ago

raw history blame contribute delete

No virus

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