ColossalAI/colossalai/engine
Jiarui Fang 681addb512
[refactor] moving grad acc logic to engine (#804)
2022-04-19 14:03:21 +08:00
..
gradient_accumulation [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +08:00
gradient_handler fix format (#586) 2022-04-06 11:40:59 +08:00
ophooks [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
paramhooks
schedule [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +08:00
__init__.py
_base_engine.py '[NFC] polish <colossalai/engine/_base_engine.py> code style' (#631) 2022-04-06 11:40:59 +08:00