HELSON f69f9bf223 [zero] add chunk init function for users (#1729) ...

* add chunk manager init function

* fix unit tests

* add comment

* add flush=True

2022-10-18 16:31:22 +08:00

..

test_chunk_mgrv2.py

[feature] A new ZeRO implementation (#1644)

2022-10-09 09:18:51 +08:00

test_chunkv2.py

[feature] A new ZeRO implementation (#1644)

2022-10-09 09:18:51 +08:00

test_fwd_bwd.py

[zero] add chunk init function for users (#1729)

2022-10-18 16:31:22 +08:00

test_optim.py

[zero] add chunk init function for users (#1729)

2022-10-18 16:31:22 +08:00

test_search.py

[zero] add chunk init function for users (#1729)

2022-10-18 16:31:22 +08:00

test_zeroddp_state_dict.py

[zero] add chunk init function for users (#1729)

2022-10-18 16:31:22 +08:00

test_zerooptim_state_dict.py

[zero] add chunk init function for users (#1729)

2022-10-18 16:31:22 +08:00