microsoft / PyCodeGPT

A pre-trained GPT model for Python code completion and generation
MIT License
263 stars 44 forks source link

Run eval on CPU instead of GPU #10

Open prasad83 opened 1 year ago

prasad83 commented 1 year ago

Use --gpu-device as -1

python eval_human_eval.py \
    --model_name_or_path PyCodeGPT-110M/ \
    --output_dir results/ \
    --num_completions 100 \
    --temperature 0.2 \
    --top_p 0.95 \
    --max_new_tokens 100 \
    --gpu_device -1