ColossalAI/colossalai/engine
Frank Lee 11f54c7b6b
[doc] improved docstring and assertion messages for the engine module (#871)
2022-04-26 10:00:18 +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 [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +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