ColossalAI/colossalai
Kai Wang (Victor Kai) b0f708dfc1 fix format (#570) 2022-04-06 11:40:59 +08:00
..
amp fix format (#570) 2022-04-06 11:40:59 +08:00
builder html refactor (#555) 2022-03-31 11:36:56 +08:00
communication [model checkpoint] updated communication ops for cpu tensors (#590) 2022-04-01 16:52:20 +08:00
context [model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
engine [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
kernel fix format (#568) 2022-04-06 11:40:59 +08:00
logging Refactored docstring to google style 2022-03-29 17:17:47 +08:00
nn fix bugs in CPU adam (#633) 2022-04-02 17:04:05 +08:00
registry Refactored docstring to google style 2022-03-29 17:17:47 +08:00
testing [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
trainer [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
utils [hotfix] fix a bug in model data stats tracing (#655) 2022-04-03 21:48:06 +08:00
zero [hotfix] fix bugs for unsharded parameters when restore data (#664) 2022-04-03 22:02:11 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
constants.py fix format constants.py (#358) 2022-03-11 15:50:28 +08:00
core.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
global_variables.py [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469) 2022-03-21 13:35:04 +08:00
initialize.py [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00