ColossalAI/tests/test_ddp
ver217 d26902645e
[ddp] add save/load state dict for ColoDDP (#1127)
* add save/load state dict for ColoDDP

* add unit test

* refactor unit test folder

* polish unit test

* rename unit test
2022-06-20 10:51:47 +08:00
..
test_ddp_ignore_params.py [ddp] add save/load state dict for ColoDDP (#1127) 2022-06-20 10:51:47 +08:00
test_ddp_state_dict.py [ddp] add save/load state dict for ColoDDP (#1127) 2022-06-20 10:51:47 +08:00