onnxruntime
b392d535 - Fix memory profiler (#14695)

Commit
2 years ago
Fix memory profiler (#14695) ### Fix memory profiler A follow up fix for PR https://github.com/microsoft/onnxruntime/pull/13495 In ORTModule training, `PartialExecuteThePlan` is called twice, we need create log event after the backward graph run complete to collect the whole training graph's activations infos. Also change some log level to verbose, to avoid too many logs in > verbose log level. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
Author
Parents
Loading