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/moe
digger yu 5e1c93d732
[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335)
9 months ago
..
__init__.py
_operation.py
checkpoint.py
experts.py
layers.py
load_balance.py
loss.py
manager.py
routers.py
utils.py