|
|
|
python run_completion.py \ |
|
--do_train \ |
|
--do_eval \ |
|
--task statement_level \ |
|
--model_name_or_path saikatc/NatGen \ |
|
--train_filename ../../../../Dataset \ |
|
--dev_filename ../../../../Dataset \ |
|
--output_dir ../../../../Saved_Models/NatGen/Code_Completion \ |
|
--beam_size 4 \ |
|
--train_batch_size 32 \ |
|
--eval_batch_size 16 \ |
|
--num_train_epochs 50 |
|
|
|
|
|
|
|
python run_completion.py \ |
|
--do_train \ |
|
--do_eval \ |
|
--task next_statement \ |
|
--model_name_or_path saikatc/NatGen \ |
|
--train_filename ../../../../Dataset \ |
|
--dev_filename ../../../../Dataset \ |
|
--output_dir ../../../../Saved_Models/NatGen/Code_Completion \ |
|
--beam_size 4 \ |
|
--train_batch_size 32 \ |
|
--eval_batch_size 16 \ |
|
--num_train_epochs 50 |