ireland-news-headlines/roberta_year_as_text_from_scratch/run.sh
2021-10-15 14:35:57 +02:00

13 lines
455 B
Bash

python run_glue.py --model_name_or_path roberta-base \
--train_file ../train/huggingface_format_year_as_text.csv \
--validation_file ../dev-0/huggingface_format_year_as_text.csv \
--do_train \
--max_seq_length 64 \
--per_device_train_batch_size 32 \
--learning_rate 2e-5 \
--num_train_epochs 3 \
--output_dir ./roberta-ireland \
--save_steps=10000 \
--eval_steps=10000 \
--evaluation_strategy steps