| 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 |