search

laekov / fastmoe

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

Update switch_gate.py #200

Closed Heihaierr closed 8 months ago

Heihaierr commented 8 months ago

fix a bug in switch_gate