ColossalAI/colossalai/engine/ophooks
Jiarui Fang 4d90a7b513
[refactor] zero directory (#724)
2022-04-11 23:13:02 +08:00
..
__init__.py [refactor] zero directory (#724) 2022-04-11 23:13:02 +08:00
_memtracer_ophook.py 'fix/format' (#573) 2022-04-06 11:40:59 +08:00
_shard_grad_ophook.py [zero] add sharded grad and refactor grad hooks for ShardedModel (#287) 2022-03-11 15:50:28 +08:00
_shard_param_ophook.py fix sharded param hook and unit test 2022-03-11 15:50:28 +08:00
utils.py [refactor] zero directory (#724) 2022-04-11 23:13:02 +08:00