ColossalAI/colossalai/engine/gradient_accumulation
Frank Lee e4685832f8
[engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)
2022-05-26 14:28:23 +08:00
..
__init__.py [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +08:00
_gradient_accumulation.py [engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030) 2022-05-26 14:28:23 +08:00