github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-25 00:24:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
5bd80b7dd1
ColossalAI/colossalai/engine
History
YuliangLiu0306 32a45cd7ef
[pipelinable]use pipelinable to support GPT model. (#903) 
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [pipelinable]use pipelinable to support GPT model.

* fix a bug caused by ShardedModel

* polish

* fix front func list
2022-05-11 09:23:58 +08:00
..
gradient_accumulation [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
gradient_handler [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
ophooks [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
paramhooks [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
schedule [pipelinable]use pipelinable to support GPT model. (#903) 2022-05-11 09:23:58 +08:00
__init__.py fix format (#580) 2022-04-06 11:40:59 +08:00
_base_engine.py [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00