0368a2665b52b843785ffe0b18b3154fd32539b0,pro_gan_pytorch/Losses.py,CondLSGAN,dis_loss,#CondLSGAN#Any#Any#Any#Any#Any#,369

Before Change

    def dis_loss(self, real_samps, fake_samps, labels, height, alpha):
        return 0.5 * (((th.mean(self.dis(real_samps, labels, height, alpha)) - 1) ** 2)
                      + (th.mean(self.dis(fake_samps, labels, height, alpha))) ** 2)

    def gen_loss(self, _, fake_samps, labels, height, alpha):
        return 0.5 * ((th.mean(self.dis(fake_samps, labels, height, alpha)) - 1) ** 2)

After Change

    def dis_loss(self, real_samps, fake_samps, labels, height, alpha):
        return 0.5 * ((th.mean((self.dis(real_samps, labels, height, alpha) - 1) ** 2))
                      + (th.mean(self.dis(fake_samps, labels, height, alpha) ** 2)))

    def gen_loss(self, _, fake_samps, labels, height, alpha):
        return 0.5 * (th.mean((self.dis(fake_samps, labels, height, alpha) - 1) ** 2))

Instances