ColossalAI/tests/test_gemini/update
HELSON a3100bd50d
[testing] add beit model for unit testings (#2196)
* [testing] add beit model

* [beit] fix bugs

* [beit] fix bugs

* [testing] fix bugs
2022-12-26 17:35:36 +08:00
..
test_chunk_mgrv2.py
test_chunkv2.py
test_convert_torch_module.py
test_fwd_bwd.py
test_gemini_use_rmt.py
test_grad_clip.py
test_optim.py [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_search.py
test_zeroddp_state_dict.py
test_zerooptim_state_dict.py