You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/nn
zbian 61e687831d
fixed using zero with tp cannot access weight correctly
2 years ago
..
_ops [hotfix]: Remove math.prod dependency (#2837) 2 years ago
layer fixed using zero with tp cannot access weight correctly 2 years ago
loss
lr_scheduler
metric
optimizer [hotfix] add correct device for fake_param (#2796) 2 years ago
parallel Rename class method of ZeroDDP (#2692) 2 years ago
__init__.py
init.py