ColossalAI/colossalai/context
アマデウス 297b8baae2
[model checkpoint] add gloo groups for cpu tensor communication (#589)
2022-04-01 10:15:52 +08:00
..
process_group_initializer [model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
random html refactor (#555) 2022-03-31 11:36:56 +08:00
__init__.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
config.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
moe_context.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
parallel_context.py [model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
parallel_mode.py [MOE] remove old MoE legacy (#493) 2022-03-22 17:37:16 +08:00
singleton_meta.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00