691e0b50d0b15665df5ec77eeec7c605c4283e1f,hypergan/trainers/alternating_trainer.py,AlternatingTrainer,_step,#AlternatingTrainer#Any#,19

Before Change

        loss = gan.loss
        metrics = gan.metrics()

        self.d_optimizer.zero_grad()

        self.before_step(self.current_step, feed_dict)

        d_loss, g_loss = self.gan.forward_loss()

After Change

        d_grads, _ = self.calculate_gradients(["d"])

        for hook in self.train_hooks:
            d_grads, _ = hook.gradients(d_grads, _)
        for p, np in zip(self.gan.d_parameters(), d_grads):
            p.grad = np

        if(len(d_grads) > 0):
                self.d_optimizer.step()

        _, g_grads = self.calculate_gradients(["g"])

        for hook in self.train_hooks:
            _, g_grads = hook.gradients(_, g_grads)
        for p, np in zip(self.gan.g_parameters(), g_grads):
            p.grad = np

        if(len(g_grads) > 0):

Instances