File size: 483 Bytes
e9af125 |
1 2 3 4 5 6 7 8 9 10 11 12 |
python -u run_generation.py \
--do_train \
--do_eval \
--do_itr \
--model_name_or_path Salesforce/codet5p-220m \
--train_filename ../../../../Dataset/Iterative_Expansion_Generation/train.jsonl \
--dev_filename ../../../../Dataset/Iterative_Expansion_Generation/valid.jsonl \
--output_dir ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Generation \
--beam_size 4 \
--train_batch_size 32 \
--eval_batch_size 16 \
--num_train_epochs 50 |