3b13f1e6b5f5ac19511cb80f8107d4a1e077beca,loss.py,GeneratorLoss,forward,#GeneratorLoss#,15

Before Change

    def forward(self, out_labels, out_images, target_images):
        // Adversarial Loss
        adversarial_loss = - out_labels.mean()
        // Perception Loss
        perception_loss = self.mse_loss(self.loss_network(out_images), self.loss_network(target_images))
        // Image Loss
        image_loss = self.mse_loss(out_images, target_images)

After Change

    def forward(self, out_labels, out_images, target_images):
        // Adversarial Loss
        adversarial_loss = torch.mean(torch.log(1 - out_labels))
        // Perception Loss
        perception_loss = self.mse_loss(self.loss_network(out_images), self.loss_network(target_images))
        // Image Loss
        image_loss = self.mse_loss(out_images, target_images)

Instances