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
Super Daniel 52d145a342
[NFC] polish colossalai/nn/lr_scheduler/onecycle.py code style (#1269)
2 years ago
..
_ops [hotfix] fix bert model test in unitests (#1272) 2 years ago
graph
layer [NFC] polish colossalai/nn/layer/parallel_sequence/_operation.py code style (#1266) 2 years ago
loss
lr_scheduler [NFC] polish colossalai/nn/lr_scheduler/onecycle.py code style (#1269) 2 years ago
metric
optimizer
parallel
__init__.py
init.py [NFC] polish colossalai/nn/init.py code style (#1292) 2 years ago