ColossalAI/colossalai/engine
YuliangLiu0306 3175bcb4d8
[pipeline]support List of Dict data (#1125)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [pipeline]support List of Dict data

* polish
2022-06-16 11:19:48 +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 [engine] fixed empty op hook check (#1096) 2022-06-10 17:27:27 +08:00
paramhooks [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
schedule [pipeline]support List of Dict data (#1125) 2022-06-16 11:19:48 +08:00
__init__.py fix format (#580) 2022-04-06 11:40:59 +08:00
_base_engine.py [engine] fixed empty op hook check (#1096) 2022-06-10 17:27:27 +08:00