ver217 8c2327b93c impl shard optim v2 and add unit test

2022-03-04 15:27:39 +08:00

..

__init__.py

[zero] add sharded grad and refactor grad hooks for ShardedModel (#287)

2022-03-02 18:28:29 +08:00

_base_ophook.py

add pytorch hooks (#179)

2022-01-25 22:20:54 +08:00

_memtracer_ophook.py

[profiler] primary memory tracer

2022-03-04 09:35:23 +08:00

_shard_grad_ophook.py

[zero] add sharded grad and refactor grad hooks for ShardedModel (#287)

2022-03-02 18:28:29 +08:00

_shard_param_ophook.py

impl shard optim v2 and add unit test

2022-03-04 15:27:39 +08:00