ColossalAI/tests/test_zero/test_mem_collector.py at e99edfcb51df48dec17498c60bbbd06baa293c22 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-03 21:23:34 +00:00

Files

Jiarui Fang 33f4412102 [Gemini] use MemStats to store the tracing data. Seperate it from Collector. (#2084 )

2022-12-06 16:43:06 +08:00

2.6 KiB

Raw Blame History

View Raw