ColossalAI/colossalai/engine
ver217 7c70bfbefa
[hotfix] fix PipelineSharedModuleGradientHandler (#1314)
2022-07-14 17:31:13 +08:00
..
gradient_accumulation
gradient_handler [hotfix] fix PipelineSharedModuleGradientHandler (#1314) 2022-07-14 17:31:13 +08:00
schedule [hotfix] remove potiential circle import (#1307) 2022-07-14 13:44:26 +08:00
__init__.py
_base_engine.py [hotfix] remove potiential circle import (#1307) 2022-07-14 13:44:26 +08:00