e427b44f4291a663b0181b8319181f470bd9d1cd,loss.py,GeneratorAdversarialWithContentLoss,forward,#GeneratorAdversarialWithContentLoss#,39

Before 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)

After Change

        content_loss = self.mse_loss(features_input, features_target)
        image_loss = self.mse_loss(out_images, target_images)
        g_tv_loss = (((out_images[:, :, :-1, :] - out_images[:, :, 1:, :]) ** 2 + (
        out_images[:, :, :, :-1] - out_images[:, :, :, 1:]) ** 2) ** 1.25).mean()
        if self.using_l1:
            // L1 Loss
            l1_loss = self.l1_loss(out_images, target_images)

Instances