unknown
add files
e9af125
raw
history blame
809 Bytes
# statement level completion
python run_completion.py \
--do_train \
--do_eval \
--task statement_level \
--model_name_or_path Salesforce/codet5-base \
--train_filename ../../../../Dataset \
--dev_filename ../../../../Dataset \
--output_dir ../../../../Saved_Models/CodeT5/Code_Completion \
--beam_size 4 \
--train_batch_size 32 \
--eval_batch_size 16 \
--num_train_epochs 50
# next statement suggestion
python run_completion.py \
--do_train \
--do_eval \
--task next_statement \
--model_name_or_path Salesforce/codet5-base \
--train_filename ../../../../Dataset \
--dev_filename ../../../../Dataset \
--output_dir ../../../../Saved_Models/CodeT5/Code_Completion \
--beam_size 4 \
--train_batch_size 32 \
--eval_batch_size 16 \
--num_train_epochs 50