ireland-news-headlines/roberta_year_as_text_better_finetunning_only_month/run.sh

2 lines
375 B
Bash

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