mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 04:29:22 +01:00
17 lines
582 B
Bash
17 lines
582 B
Bash
PYTHONPATH=. torchrun examples/pytorch/text_generation/finetune_text_generation.py \
|
|
--trainer 'text-generation-trainer' \
|
|
--work_dir './tmp' \
|
|
--task 'text2text-generation' \
|
|
--model 'damo/nlp_mt5_zero-shot-augment_chinese-base' \
|
|
--train_dataset_name 'DuReader_robust-QG' \
|
|
--val_dataset_name 'DuReader_robust-QG' \
|
|
--train_split 'train' \
|
|
--val_split 'validation' \
|
|
--src_txt 'text1' \
|
|
--tgt_txt 'text2' \
|
|
--max_epochs 1 \
|
|
--use_model_config True \
|
|
--per_device_train_batch_size 8 \
|
|
--lr 1e-3 \
|
|
--lr_scheduler 'noam' \
|