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

Before Change

    def forward(self, x):
        out, idx = self.mobile_inverted_conv(x)
        // TODO: unify idx format
        if not isinstance(idx, int):
            idx = (idx == 1).nonzero()
        if self.op_candidates_list[idx].is_zero_layer():
            res = x
        elif self.shortcut is None:
            res = out

After Change

            // is zero layer
            return x
        if self.shortcut is None:
            return out
        return out + self.shortcut(x)


class ShuffleLayer(nn.Module):

Instances