ireland-news-headlines/t5_no_year/run.sh

20 lines
648 B
Bash
Raw Normal View History

2021-10-01 13:18:24 +02:00
python run_summarization.py \
2021-10-03 13:05:53 +02:00
--model_name_or_path google/t5-v1_1-base \
2021-10-01 13:18:24 +02:00
--do_train \
--train_file ../train/huggingface_format.csv \
2021-10-03 13:05:53 +02:00
--validation_file ../dev-0/huggingface_format_short.csv \
2021-10-01 13:18:24 +02:00
--source_prefix "classify: " \
--summary_column 'label' \
--max_target_length=4 \
--max_source_length=64 \
2021-10-04 17:28:22 +02:00
--num_train_epochs 20 \
2021-10-01 13:18:24 +02:00
--output_dir ./t5-retrained \
--per_device_train_batch_size=16 \
--per_device_eval_batch_size=16 \
2021-10-03 13:05:53 +02:00
--gradient_accumulation_steps=4 \
2021-10-01 13:18:24 +02:00
--overwrite_output_dir \
--predict_with_generate \
--save_steps=10000 \
2021-10-03 13:05:53 +02:00
--eval_steps=10000 \
--evaluation_strategy steps