[PyTorch] Fix NaN expert weight gradients at num_groups == 1 with SReLU - #3485
Open
GarlGuo wants to merge 1 commit into
Open
[PyTorch] Fix NaN expert weight gradients at num_groups == 1 with SReLU#3485GarlGuo wants to merge 1 commit into
GarlGuo wants to merge 1 commit into
