ADD file via upload
This commit is contained in:
parent
13b306e5dc
commit
32dd546133
|
|
@ -0,0 +1,18 @@
|
|||
python linevul_main_predict_test.py \
|
||||
--output_dir=./saved_models \
|
||||
--model_type=roberta \
|
||||
--model_name=model.bin \
|
||||
--tokenizer_name=microsoft/codebert-base \
|
||||
--model_name_or_path=microsoft/codebert-base \
|
||||
--do_test \
|
||||
--train_data_file=../data/big-vul_dataset/new_train.csv \
|
||||
--eval_data_file=../data/big-vul_dataset/new_valid.csv \
|
||||
--test_data_file=../data/big-vul_dataset/test.csv \
|
||||
--epochs 10 \
|
||||
--block_size 512 \
|
||||
--train_batch_size 16 \
|
||||
--eval_batch_size 16 \
|
||||
--learning_rate 2e-5 \
|
||||
--max_grad_norm 1.0 \
|
||||
--evaluate_during_training \
|
||||
--seed 123456
|
||||
Loading…
Reference in New Issue