ireland-news-headlines/roberta_year_as_token_better_finetunning/run.sh
kubapok aa2eac49c3 a
2022-02-10 14:37:36 +01:00

2 lines
386 B
Bash

python run_glue.py --model_name_or_path roberta-base --train_file ../train/huggingface_format_year.csv --validation_file ../dev-0/huggingface_format_year.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