Baizhou Zhang d99b2c961a [hotfix] fix grad accumulation plus clipping for gemini (#5002)

2023-11-02 17:59:10 +08:00

..

test_chunk_mgrv2.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

test_chunkv2.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

test_fwd_bwd.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_gemini_use_rmt.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_grad_accum.py

[hotfix] fix grad accumulation plus clipping for gemini (#5002)

2023-11-02 17:59:10 +08:00

test_grad_clip.py

[hotfix] fix grad accumulation plus clipping for gemini (#5002)

2023-11-02 17:59:10 +08:00

test_inference.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_optim.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_runtime_mem_tracer.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_search.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_zeroddp_state_dict.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

test_zerooptim_state_dict.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00