tests/test_moe/test_top1.py · 219df6e685c7ed1d0b712ce673debc8684bc3779 · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

Optimized MoE layer and fixed some bugs; · 219df6e6
1SAA authored Feb 18, 2022
```
Decreased moe tests;

Added FFNExperts and ViTMoE model
```
219df6e6

test_top1.py 3.27 KB