Views
No views yet
1lang=python
2mkdir -p ./saved_models/$lang
3python run.py \
4 --output_dir=./saved_models/$lang \
5 --config_name=microsoft/graphcodebert-base \
6 --model_name_or_path=microsoft/graphcodebert-base \
7 --tokenizer_name=microsoft/graphcodebert-base \
8 --lang=$lang \
9 --do_train \
10 --train_data_file=dataset/$lang/train.jsonl \
11 --eval_data_file=dataset/$lang/valid.jsonl \
12 --test_data_file=dataset/$lang/test.jsonl \
13 --codebase_file=dataset/$lang/codebase.jsonl \
14 --num_train_epochs 10 \
15 --code_length 256 \
16 --data_flow_length 64 \
17 --nl_length 128 \
18 --train_batch_size 16 \
19 --eval_batch_size 32 \
20 --learning_rate 2e-5 \
21 --seed 123456 2>&1| tee saved_models/$lang/train.log