f7f68faa11c4a1280a17f5482f11c46606653b77,hypergan/train_hooks/adversarial_norm_train_hook.py,AdversarialNormTrainHook,forward,#AdversarialNormTrainHook#,21

Before Change

          self.target.data = self.gan.x.data.clone()
          loss, norm = self.gan.regularize_adversarial_norm(self.gan.discriminator(self.target), self.gan.d_fake, self.target)
      elif self.config.target == "g":
          self.target.data = self.gan.g.data.clone()
          loss, norm = self.gan.regularize_adversarial_norm(self.gan.d_real, self.gan.discriminator(self.target), self.target)

      if loss is None:
          return [None, None]

After Change

    def forward(self):
        if self.config.mode == "real":
            for target, data in zip(self.target, self.gan.discriminator_real_inputs()):
                target.data = data.clone()
            d_fake = self.gan.d_fake
            d_real = self.gan.forward_discriminator(self.target)
        elif self.config.mode == "fake":
            for target, data in zip(self.target, self.gan.discriminator_fake_inputs()):

Instances