ColossalAI/colossalai/trainer
Jie Zhu 73d36618a6
[profiler] add MemProfiler (#356)
* add memory trainer hook

* fix bug

* add memory trainer hook

* fix import bug

* fix import bug

* add trainer hook

* fix #370 git log bug

* modify `to_tensorboard` function to support better output

* remove useless output

* change the name of `MemProfiler`

* complete memory profiler

* replace error with warning

* finish trainer hook

* modify interface of MemProfiler

* modify `__init__.py` in profiler

* remove unnecessary pass statement

* add usage to doc string

* add usage to trainer hook

* new location to store temp data file
2022-03-29 12:48:34 +08:00
..
hooks [profiler] add MemProfiler (#356) 2022-03-29 12:48:34 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_trainer.py add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00