ireland-news-headlines/roberta_no_year_from_scratch/run.sh

13 lines
465 B
Bash
Raw Normal View History

python run_glue.py --model_name_or_path roberta-base \
--train_file ../train/huggingface_format.csv \
--validation_file /media/kuba/ssdsam/gonito/ireland-news/dev-0/huggingface_format.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