ColossalAI/colossalai/nn/loss/__init__.py

2 lines
53 B
Python

# from .loss_moe import MoeCrossEntropyLoss, MoeLoss