File size: 743 Bytes
e9af125 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# statement level completion
python run_completion.py \
--do_test \
--task statement_level \
--model_name_or_path MickyMike/codebert-c \
--load_model_path ../../../../Saved_Models/CodeBert/Code_Completion \
--test_filename ../../../../Dataset \
--output_dir ../../../../Saved_Models/CodeBert/Code_Completion \
--beam_size 4 \
--eval_batch_size 16
# next statement suggestion
python run_completion.py \
--do_test \
--task next_statement \
--model_name_or_path MickyMike/codebert-c \
--load_model_path ../../../../Saved_Models/CodeBert/Code_Completion \
--test_filename ../../../../Dataset \
--output_dir ../../../../Saved_Models/CodeBert/Code_Completion \
--beam_size 4 \
--eval_batch_size 16 |