f3e7b16ce89d4e737e7608996df2bd241d594a58,se3cnn/blocks/gated_block.py,GatedBlock,forward,#GatedBlock#,96

Before Change

        // convolution
        z = self.conv(x)

        // gate
        if self.scalar_act is not None or self.gate_act is not None:
            z = torch.utils.checkpoint.checkpoint(gate, z) if self.checkpoint else gate(z)

        // stride
        if self.stride > 1:
            z = low_pass_filter(z, self.stride, self.stride)

        // dropout
        if self.dropout is not None:
            z = self.dropout(z)

        return z

After Change

        if self.gate_act is not None:
            g = y[:, size_out:]
            assert g.size(1) == sum(self.repr_out[1:])
            g = self.gate_act(g)
            begin_g = 0  // index of first scalar gate capsule

        z = y.new_empty(batch, size_out, *size)

Instances