onnxruntime/orttraining
Wei-Sheng Chin 189fb60ef9
Fix a bug and add code to profile memory (#4241)
* Fix a bug and add code to profile memory

1. Compile Send/Recv again (currently broken because of
   HOROVOD refactor).
2. Add code to print out initializer allocation size and
   activation memory size.

* Address comments

* Split memory counts per locations

* Fix a metric
2020-06-16 10:17:27 -07:00
..
orttraining Fix a bug and add code to profile memory (#4241) 2020-06-16 10:17:27 -07:00
pytorch_frontend_examples Remove ORTModel from frontend API (#3825) 2020-05-11 18:20:33 -07:00
tools Use azcopy to download test data (#4221) 2020-06-16 10:14:34 -07:00