ColossalAI/colossalai
ver217 7d3711058f
fix zero3 fp16 and add zero3 model context (#62)
2021-12-10 17:48:50 +08:00
..
amp Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
builder Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
communication Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
context Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
engine Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
logging Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
nn Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
registry Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
trainer update markdown docs (english) (#60) 2021-12-10 14:37:33 +08:00
utils update markdown docs (english) (#60) 2021-12-10 14:37:33 +08:00
zero fix zero3 fp16 and add zero3 model context (#62) 2021-12-10 17:48:50 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
constants.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
core.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
initialize.py fix zero3 fp16 and add zero3 model context (#62) 2021-12-10 17:48:50 +08:00