ColossalAI/colossalai/utils/memory_tracer
ver217 ab8c6b4a0e
[zero] refactor memstats collector (#706)
* refactor memstats collector

* fix disposable

* polish code
2022-04-11 10:46:08 +08:00
..
__init__.py [zero] initialize a stateful tensor manager (#614) 2022-04-06 16:18:49 +08:00
async_memtracer.py polish utils docstring (#620) 2022-04-01 16:36:47 +08:00
memstats_collector.py [zero] refactor memstats collector (#706) 2022-04-11 10:46:08 +08:00
model_data_memtracer.py [hotfix] fix a bug in model data stats tracing (#655) 2022-04-03 21:48:06 +08:00