| # do test | |
| python ./Scripts/UnixCoder/run_one_model.py \ | |
| --do_function_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 | |