intelligent-machine-learning/dlrover

Add balance loss in atorch moe example

Open

#1.300 geöffnet am 18. Okt. 2024

Auf GitHub ansehen
 (0 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (212 Forks)github user discovery
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.

Contributor Guide