File size: 351 Bytes
e9af125
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
python -u run_generation.py \
    --do_test \
    --model_name_or_path Salesforce/codet5p-220m \
    --load_model_path ../../../../Saved_Models/CodeT5+/Code_Generation \
    --test_filename ../../../../Dataset/Code_Generation/test.jsonl \
    --output_dir ../../../../Saved_Models/CodeT5+/Code_Generation \
    --beam_size 4 \
    --eval_batch_size 16