intelligent-machine-learning/dlrover
Auf GitHub ansehenAdd balance loss in atorch moe example
Open
#1.300 geöffnet am 18. Okt. 2024
Hacktoberfesttodo
Repository-Metriken
- Stars
- (1.662 Stars)
- PR-Merge-Metriken
- (PR-Metriken ausstehend)
Beschreibung
Add balance loss in moe example.
We add use aux loss to router in moe example, similar to aux loss implementation in megatron moe.