github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/engine
History
YuliangLiu0306 b167258b6a
[pipeline]refactor ppschedule to support tensor list (#1050) 
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* refactor ppschedule to support tensor list

* polish
 		2022-06-02 13:48:59 +08:00
..
gradient_accumulation [engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030) 2022-05-26 14:28:23 +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 [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00
__init__.py
…
_base_engine.py [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00