File size: 685 Bytes
1455619 d816f2e 1455619 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#! /usr/bin/env bash
python run_summarization_flax.py \
--output_dir ./ \
--model_name_or_path google/mt5-small \
--tokenizer_name google/mt5-small \
--train_file qgen_train.csv \
--validation_file qgen_val.csv \
--text_column context \
--summary_column question \
--do_train \
--do_eval \
--num_train_epochs 3 \
--eval_steps 2000 \
--logging_steps 100 \
--learning_rate 5e-5 --warmup_steps 1000 \
--per_device_train_batch_size 2 \
--per_device_eval_batch_size 2 \
--overwrite_output_dir \
--max_source_length 1024 --max_target_length 64 \
--push_to_hub
|