File size: 1,653 Bytes
e9af125
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# statement level completion
python run_completion.py \
	--do_test \
    --task statement_level \
    --do_itr \
    --do_cpuonly \
    --model_name_or_path Salesforce/codet5p-220m \
 	--load_model_path ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
    --beam_size 4 \
    --eval_batch_size 16

python run_completion.py \
	--do_test \
    --task statement_level \
    --do_itr \
    --model_name_or_path Salesforce/codet5p-220m \
 	--load_model_path ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
    --beam_size 4 \
    --eval_batch_size 16


# next statement suggestion
python run_completion.py \
	--do_test \
    --task next_statement \
    --do_itr \
    --do_cpuonly \
    --model_name_or_path Salesforce/codet5p-220m \
 	--load_model_path ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
    --beam_size 4 \
    --eval_batch_size 16


python run_completion.py \
	--do_test \
    --task next_statement \
    --do_itr \
    --model_name_or_path Salesforce/codet5p-220m \
 	--load_model_path ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
 	--test_filename ../../../../Dataset \
    --output_dir ../../../../Saved_Models/CodeT5+/Iterative_Expansion_Completion \
    --beam_size 4 \
    --eval_batch_size 16