optimum
eada1577 - ORTTrainer / ORTSeq2SeqTrainer refactoring - Inference with ORTModel (#189)

Commit
3 years ago
ORTTrainer / ORTSeq2SeqTrainer refactoring - Inference with ORTModel (#189) * Inference with ORTModel * Clean up unused imports * Replace Inference session by ort model * Inference with ORTModel * Replace in evaluation_loop * refactoring prediction_loop * ORTSeq2SeqTrainer refactoring - Inference with ORTModel (#359) * Override export of ORTSeq2SeqTrainer * Update OnnxConfigWithLoss wrapper * Fix ORTTrainer inference ort subclass parsing * Override the evaluation and prediction loop in ORTSeq2SeqTrainer * detect labels from input names * Detect loss from output names * Put back decoder with past * Put past key values in the correct place * remove if/else statement * Revert the inference code
Author
Parents
Loading