ColossalAI/colossalai/nn/parallel
Ziyue Jiang 4fc748f69b
[Tensor] fix optimizer for CPU parallel (#1069)
2022-06-06 17:36:11 +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 [Tensor] fix optimizer for CPU parallel (#1069) 2022-06-06 17:36:11 +08:00