megablocks
megablocks copied to clipboard
fix eval for moe layer
when I run /exp/dmoe/dmoe_46m_8gpu.sh,I encountered the following Error during evaluation.
the func save_load_balancing_loss is bypassed during evaluation.