ColossalAI/colossalai/nn
Jiarui Fang 49832b2344
[refactory] add nn.parallel module (#1068)
2022-06-06 15:34:41 +08:00
..
_ops
layer
loss
lr_scheduler
metric
model
optimizer
parallel [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
init.py