dafdc9b5cb8fa4c65285aad22b1429549d06d71a,networks/correlation_package/modules/correlation.py,Correlation,forward,#Correlation#,15

Before Change

    def forward(self, input1, input2):

        result = CorrelationFunction(self.pad_size, self.kernel_size, self.max_displacement,self.stride1, self.stride2, self.corr_multiply)(input1, input2)

        return result

After Change

        self.corr_multiply = corr_multiply

    def forward(self, input1, input2):
        return CorrelationFunction.apply(input1, input2, self.pad_size,
                                         self.kernel_size,
                                         self.max_displacement, self.stride1,
                                         self.stride2, self.corr_multiply)

Instances