cf2d46f50b55271fc4401a21bbf87cd59f6f4d97,gluon/models/shufflenet.py,ShuffleUnit,hybrid_forward,#ShuffleUnit#Any#Any#,98

Before Change

    def hybrid_forward(self, F, x):
        identity = x
        x = self.activ(self.compress_bn1(self.compress_conv1(x)))
        x = self.c_shuffle(x)
        x = self.dw_bn2(self.dw_conv2(x))
        x = self.expand_bn3(self.expand_conv3(x))
        if self.downsample:
            identity = self.avgpool(identity)
            x = F.concat(x, identity, dim=1)
        else:

After Change

    def hybrid_forward(self, F, x):
        identity = x
        x = self.compress_conv1(x)
        x = self.compress_bn1(x)
        x = self.activ(x)
        x = self.c_shuffle(x)
        x = self.dw_conv2(x)
        x = self.dw_bn2(x)
        x = self.expand_conv3(x)
        x = self.expand_bn3(x)
        if self.downsample:
            identity = self.avgpool(identity)
            x = F.concat(x, identity, dim=1)
        else:

Instances