ColossalAI/colossalai/zero
Jiarui Fang e99edfcb51
[NFC] polish comments for Chunk class (#2116)
2022-12-12 15:39:31 +08:00
..
init_ctx
shard_utils
sharded_model [Gemini] use MemStats to store the tracing data. Seperate it from Collector. (#2084) 2022-12-06 16:43:06 +08:00
sharded_optim [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +08:00
sharded_param
utils [NFC] polish comments for Chunk class (#2116) 2022-12-12 15:39:31 +08:00
__init__.py [Gemini] add GeminiAdamOptimizer (#1960) 2022-11-16 14:44:28 +08:00