ColossalAI/colossalai
Jiarui Fang 2c0c85d3d3
fix a bug in timer (#114)
2022-01-05 16:07:06 +08:00
..
amp Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
builder Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
communication Layer integration (#83) 2021-12-27 15:04:32 +08:00
context update default logger (#100) (#101) 2022-01-04 20:03:26 +08:00
engine fix layers/schedule for hybrid parallelization (#111) (#112) 2022-01-04 20:52:31 +08:00
kernel add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
logging update default logger (#100) (#101) 2022-01-04 20:03:26 +08:00
nn fix layers/schedule for hybrid parallelization (#111) (#112) 2022-01-04 20:52:31 +08:00
registry Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
trainer fix a bug in timer (#114) 2022-01-05 16:07:06 +08:00
utils Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
zero fixed zero level 3 dtype bug (#76) 2021-12-20 17:00:53 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
constants.py Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
core.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
initialize.py Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00