Kai Wang (Victor Kai) 50f2ad213f [NFC] polish colossalai/engine/ophooks/utils.py code style (#1256)

2022-07-13 12:08:21 +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

[NFC] polish colossalai/engine/ophooks/utils.py code style (#1256)

2022-07-13 12:08:21 +08:00

paramhooks

[doc] improved docstring and assertion messages for the engine module (#871)

2022-04-26 10:00:18 +08:00

schedule

[hotfix] fix an assertion bug in base schedule. (#1250)

2022-07-12 14:20:02 +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