24d6018ac120b75d91d94f2a238ac9f1c0bfbd71,hypergan/gans/aligned_interpolate_gan.py,AlignedInterpolateGAN,forward_discriminator,#AlignedInterpolateGAN#Any#,39

Before Change

        self.gammas = [torch.Tensor([self.config.interpolate]).float()[0].cuda(), torch.Tensor([1.-self.config.interpolate]).float()[0].cuda()]

    def forward_discriminator(self, inputs):
        return self.discriminator(inputs[0])*self.gammas[0] + self.discriminator(inputs[1]) * self.gammas[1]

    def forward_pass(self):
        self.x = self.inputs.next()

After Change

        self.gammas = [torch.Tensor([self.config.interpolate]).float()[0].cuda(), torch.Tensor([1.-self.config.interpolate]).float()[0].cuda()]

    def forward_discriminator(self, inputs):
        d0 = self.discriminator(inputs[0])
        d1 = self.discriminator2(inputs[1])
        return self.sigmoid(d1)*d0*self.gammas[0]  + self.sigmoid(d0)*d1*self.gammas[1]

    def forward_pass(self):
        self.x = self.inputs.next()

Instances