You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
fanjinfucool ffad81e1d1
fix format (#585)
3 years ago
..
amp fix format (#570) 3 years ago
builder html refactor (#555) 3 years ago
communication [model checkpoint] updated communication ops for cpu tensors (#590) 3 years ago
context [model checkpoint] add gloo groups for cpu tensor communication (#589) 3 years ago
engine fix format (#585) 3 years ago
kernel fix format (#583) 3 years ago
logging Refactored docstring to google style 3 years ago
nn fix bugs in CPU adam (#633) 3 years ago
registry Refactored docstring to google style 3 years ago
testing [test] fixed rerun_on_exception and adapted test cases (#487) 3 years ago
trainer [refactor] pipeline, put runtime schedule into engine. (#627) 3 years ago
utils [hotfix] fix a bug in model data stats tracing (#655) 3 years ago
zero [hotfix] fix bugs for unsharded parameters when restore data (#664) 3 years ago
__init__.py Develop/experiments (#59) 3 years ago
constants.py fix format constants.py (#358) 3 years ago
core.py [polish] polish singleton and global context (#500) 3 years ago
global_variables.py [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469) 3 years ago
initialize.py [refactor] pipeline, put runtime schedule into engine. (#627) 3 years ago