ADD file via upload

This commit is contained in:
zhongfa1999 2024-10-21 17:41:34 +08:00
parent 13b306e5dc
commit 32dd546133
1 changed files with 18 additions and 0 deletions

View File

@ -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