optimum
a1dd179d - [ORTTrainer] Fix to be compatible with onnxruntime 1.11.0 (#124)

Commit
3 years ago
[ORTTrainer] Fix to be compatible with onnxruntime 1.11.0 (#124) * Fix compatibility of run_ner with gpt2 * Modify to be compatible with ort 1.11.0 * Workaround evaluate for ORTTrainer * Add inference_with_ort flag * Fix case: predict_with_generate + Inference_with_ort * raise NotImplementedError Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> * Set loss of ORT to None
Author
Parents
Loading