File size: 1,132 Bytes
51c57f8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
lang=$1
current_time=$(date "+%Y%m%d%H%M%S")
code_length=256
nl_length=128
model_type=cocosoda #"base", "cocosoda" ,"multi-loss-cocosoda"
moco_k=1024
moco_m=0.999
lr=2e-5
moco_t=0.07
max_steps=1000
aug_type_way=random_replace_type 
data_aug_type=random_mask
base_model=DeepSoftwareAnalytics/CoCoSoDa
CUDA_VISIBLE_DEVICES=0

function zero-shot () {
output_dir=./saved_models/zero-shot/${lang}
mkdir -p $output_dir

 CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES} python run.py   --eval_frequency  100 \
    --do_zero_short \
    --moco_m ${moco_m} --moco_t ${moco_t}  \
    --model_type ${model_type} \
    --output_dir ${output_dir}  \
    --data_aug_type ${data_aug_type} \
    --moco_k ${moco_k} \
    --config_name=${base_model}  \
    --model_name_or_path=${base_model} \
    --tokenizer_name=${base_model} \
    --lang=$lang \
    --do_test \
    --test_data_file=dataset/$lang/test.jsonl \
    --codebase_file=dataset/$lang/codebase.jsonl \
    --code_length ${code_length} \
    --nl_length ${nl_length} \
    --eval_batch_size 128 \
    --learning_rate ${lr} \
    --seed 123456 2>&1| tee ${output_dir}/running.log
}
zero-shot