ColossalAI/colossalai
Jiang Zhuo 5a4a3b77d9 fix format (#376) 2022-03-11 15:50:28 +08:00
..
amp fix format (#362) 2022-03-11 15:50:28 +08:00
builder
communication fix format (#332) 2022-03-11 15:50:28 +08:00
context flake8 style change (#363) 2022-03-11 15:50:28 +08:00
engine Fix/format colossalai/engine/paramhooks/(#350) 2022-03-11 15:50:28 +08:00
kernel
logging
nn fix format (#376) 2022-03-11 15:50:28 +08:00
registry
trainer
utils Added activation offload (#331) 2022-03-11 15:50:28 +08:00
zero [bug] shard param during initializing the ShardedModelV2 (#381) 2022-03-11 15:50:28 +08:00
__init__.py
constants.py fix format constants.py (#358) 2022-03-11 15:50:28 +08:00
core.py
global_variables.py
initialize.py