dd9512019c67fdb06c0e53991608d36ca5cdf920,loss.py,GeneratorAdversarialWithContentLoss,forward,#GeneratorAdversarialWithContentLoss#Any#Any#Any#,38

Before Change

    def forward(self, out_labels, out_images, target_images):
        // Adversarial Loss
        // adversarial_loss = torch.mean(torch.log(1 - out_labels))
        adversarial_loss = -out_labels.mean()
        // Content Loss
        features_input = self.loss_network(out_images)
        features_target = self.loss_network(target_images)
        content_loss = self.mse_loss(features_input, features_target)

After Change

    def forward(self, out_labels, out_images, target_images):
        // Adversarial Loss
        adversarial_loss = torch.mean(torch.log(1 - out_labels))
        // Content Loss
        features_input = self.loss_network(out_images)
        features_target = self.loss_network(target_images)
        content_loss = self.mse_loss(features_input, features_target)

Instances