You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/engine
Jiarui Fang 681addb512
[refactor] moving grad acc logic to engine (#804)
3 years ago
..
gradient_accumulation [refactor] moving grad acc logic to engine (#804) 3 years ago
gradient_handler fix format (#586) 3 years ago
ophooks [refactor] moving memtracer to gemini (#801) 3 years ago
paramhooks Fix/format colossalai/engine/paramhooks/(#350) 3 years ago
schedule [refactor] moving grad acc logic to engine (#804) 3 years ago
__init__.py fix format (#580) 3 years ago
_base_engine.py '[NFC] polish <colossalai/engine/_base_engine.py> code style' (#631) 3 years ago