ColossalAI/colossalai/engine
Jie Zhu 3213554cc2 [profiler] add adaptive sampling to memory profiler (#330)
* fix merge conflict

modify unit test

remove unnessesary log info

reformat file

* remove unused module

* remove unnecessary sync function

* change doc string style from Google to Sphinx
2022-03-11 15:50:28 +08:00
..
gradient_handler Feature/zero (#279) 2022-03-11 15:50:28 +08:00
ophooks [profiler] add adaptive sampling to memory profiler (#330) 2022-03-11 15:50:28 +08:00
paramhooks add a common util for hooks registered on parameter. (#292) 2022-03-11 15:50:28 +08:00
schedule Feature/zero (#279) 2022-03-11 15:50:28 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_base_engine.py [profiler] primary memory tracer 2022-03-11 15:50:28 +08:00