ColossalAI/colossalai
Frank Lee af88570f4b fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes 2021-11-10 10:47:58 +08:00
..
builder Add gradient accumulation, fix lr scheduler 2021-11-08 16:30:24 +08:00
communication Migrated project 2021-10-28 18:21:23 +02:00
context Migrated project 2021-10-28 18:21:23 +02:00
engine fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes 2021-11-10 10:47:58 +08:00
logging Migrated project 2021-10-28 18:21:23 +02:00
nn fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes 2021-11-10 10:47:58 +08:00
registry Migrated project 2021-10-28 18:21:23 +02:00
trainer fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes 2021-11-10 10:47:58 +08:00
utils fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes 2021-11-10 10:47:58 +08:00
__init__.py Migrated project 2021-10-28 18:21:23 +02:00
checkpointing.py Migrated project 2021-10-28 18:21:23 +02:00
constants.py Migrated project 2021-10-28 18:21:23 +02:00
core.py Migrated project 2021-10-28 18:21:23 +02:00
initialize.py Add gradient accumulation, fix lr scheduler 2021-11-08 16:30:24 +08:00