ColossalAI/colossalai/nn/parallel at bfdc5ccb7b68937e2a1dfd358e9b62b06fbe40ea - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

Ziyue Jiang 4fc748f69b [Tensor] fix optimizer for CPU parallel (#1069 )

2022-06-06 17:36:11 +08:00

..

[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