python run_generation.py \ | |
--do_train \ | |
--do_eval \ | |
--model_name_or_path MickyMike/codebert-c \ | |
--train_filename ../../../../Dataset/Code_Generation/train.jsonl \ | |
--dev_filename ../../../../Dataset/Code_Generation/valid.jsonl \ | |
--output_dir ../../../../Saved_Models/CodeBert/Code_Generation \ | |
--beam_size 4 \ | |
--train_batch_size 32 \ | |
--eval_batch_size 16 \ | |
--num_train_epochs 50 | |