1e39951b371238b26783b31dd5101eb334c23412,kraken/lib/layers.py,ActConv2D,resize,#ActConv2D#,703

Before Change

        idx = torch.tensor([x for x in range(old_shape) if x not in del_indices])
        weight = self.co.weight.index_select(0, idx)
        rweight = torch.zeros((output_size - weight.size(0), weight.size(1), weight.size(2), weight.size(3)))
        torch.nn.init.xavier_uniform_(rweight)
        weight = torch.cat([weight, rweight], dim=0)
        bias = self.co.bias.index_select(0, idx)
        bias = torch.cat([bias, torch.zeros(output_size - bias.size(0))])
        self.co = torch.nn.Conv2d(self.in_channels, self.out_channels, self.kernel_size,

After Change

        idx = torch.tensor([x for x in range(old_shape) if x not in del_indices])
        weight = self.co.weight.index_select(0, idx)
        rweight = torch.zeros((output_size - weight.size(0), weight.size(1), weight.size(2), weight.size(3)))
        if rweight.shape[0] > 0:
            torch.nn.init.xavier_uniform_(rweight)
        weight = torch.cat([weight, rweight], dim=0)
        bias = self.co.bias.index_select(0, idx)
        bias = torch.cat([bias, torch.zeros(output_size - bias.size(0))])
        self.co = torch.nn.Conv2d(self.in_channels, self.out_channels, self.kernel_size,

Instances