ColossalAI/colossalai/gemini/memory_tracer at bfdc5ccb7b68937e2a1dfd358e9b62b06fbe40ea - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-27 02:03:13 +00:00

Files

History

ver217 d7e0303d1e [zero] use GeminiMemoryManager when sampling model data (#850 )

2022-04-24 17:17:22 +08:00

..

__init__.py

[refactor] moving memtracer to gemini (#801 )

2022-04-19 10:13:08 +08:00

memory_monitor.py

[refactor] moving memtracer to gemini (#801 )

2022-04-19 10:13:08 +08:00

memstats_collector.py

[zero] use GeminiMemoryManager when sampling model data (#850 )

2022-04-24 17:17:22 +08:00

model_data_memtracer.py

[refactor] moving memtracer to gemini (#801 )

2022-04-19 10:13:08 +08:00