onnxruntime/orttraining
Vincent Wang a2658f0784
[ORTModule] Fix Graph Builder for Eval Mode (#13255)
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.
2022-10-12 14:39:54 +08:00
..
orttraining [ORTModule] Fix Graph Builder for Eval Mode (#13255) 2022-10-12 14:39:54 +08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools [ROCm] Update ROCm and MIGraphX CI pipeline to ROCm5.3 (#13257) 2022-10-11 13:47:22 +08:00