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 |