mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-20 19:12:24 +00:00
Current graph builder for ORTModule will apply the training's graph optimizations for both training and eval mode. Take BatchNorm as example, one of training's graph optimizations will replace BatchNormalization Op to BatchNormInternal which is for training only. This PR is to fix this, for eval mode, we will not apply the training's graph optimizations. The inference's graph optimizations will be applied when InferenceSession initialization. |
||
|---|---|---|
| .. | ||
| orttraining | ||
| pytorch_frontend_examples | ||
| tools | ||