File size: 773 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/graphcodebert-c \
 	--load_model_path ../../../../Saved_Models/GraphCodeBert/Code_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/GraphCodeBert/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/graphcodebert-c \
 	--load_model_path ../../../../Saved_Models/GraphCodeBert/Code_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/GraphCodeBert/Code_Completion \
    --beam_size 4 \
    --eval_batch_size 16