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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 10:54:49 +00:00

Files

History

HELSON b28991dd0a [feature] A new ZeRO implementation (#1644 )

2022-10-09 09:18:51 +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

[feature] A new ZeRO implementation (#1644 )

2022-10-09 09:18:51 +08:00

model_data_memtracer.py

[refactor] moving memtracer to gemini (#801 )

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