5dec8de970eb520f6d4e6cc6c7934eb564a42a6c,nussl/ml/networks/modules.py,InstanceNorm,forward,#InstanceNorm#Any#,30

Before Change

    def forward(self, data):
        if self.use_instance_norm:
            shape = data.shape
            data = data.view(shape[0], 1, shape[1], -1)
            data = self.instance_norm(data)
            data = data.view(shape)
        return data

After Change

    
    def forward(self, data):
        if self.use_instance_norm:
            data = data.transpose(2, 1)
            shape = data.shape
            new_shape = (shape[0], self.num_features, -1)

            data = data.view(new_shape)
            data = self.instance_norm(data)
            data = data.view(shape)

Instances