File size: 428 Bytes
6705032 0cea2c9 6705032 0cea2c9 1d488b6 6705032 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# do test
python ./Scripts/UnixCoder/run_one_model.py \
--do_test \
--model_name_or_path ../../models/UnixCoder \
--test_filename ../../dataset/test.jsonl \
--output_dir ../../models/FT_Model \
--beam_size 1 \
--train_batch_size 256 \
--eval_batch_size 256 \
--learning_rate 6e-5 \
--gradient_accumulation_steps 2 \
--num_train_epochs 10 \
--mse_loss_weight 0.9 \
--ce_loss_weight 0.1
|