ColossalAI/colossalai/nn/parallel
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
..
layers [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
__init__.py [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
data_parallel.py [ddp] add save/load state dict for ColoDDP (#1127) 2022-06-20 10:51:47 +08:00