search

laekov / fastmoe

A fast MoE impl for PyTorch
https://fastmoe.ai
Apache License 2.0
1.52k stars 184 forks source link

Update switch_gate.py #200

Closed Heihaierr closed 6 months ago

Heihaierr commented 6 months ago

fix a bug in switch_gate