Jiarui Fang b334822163 [zero] polish sharded param name (#484) ...

* [zero] polish sharded param name

* polish code

* polish

* polish code

* polish

* polsih

* polish

2022-03-22 14:36:16 +08:00

..

test_activation_checkpointing.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

test_bucket_tensor_copy.py

[zero] polish sharded param name (#484)

2022-03-22 14:36:16 +08:00

test_gradient_accumluation.py

[test] optimized zero data parallel test (#452)

2022-03-18 11:35:54 +08:00

test_zero_gradient_clippling.py

[test] optimized zero data parallel test (#452)

2022-03-18 11:35:54 +08:00