165756cc19a19db0e75c128ccc97eff3579af1c3,examples/nas/proxylessnas/ops.py,MobileInvertedResidualBlock,forward,#MobileInvertedResidualBlock#Any#,41

Before Change

            res = out
        else:
            conv_x = out
            skip_x = self.shortcut(x)
            res = skip_x + conv_x
        return res

After Change

        if torch.sum(torch.abs(out)).item() == 0 and x.size() == out.size():
            // is zero layer
            return x
        if self.shortcut is None:
            return out
        return out + self.shortcut(x)


class ShuffleLayer(nn.Module):
    def __init__(self, groups):

Instances