fb5a8a3e0a73eea528ac398c04ba20a96ef575e1,examples/experimental/next-frame.py,NextFrameGAN,d_parameters,#NextFrameGAN#,117

Before Change

            yield param

    def d_parameters(self):
        return self.discriminator.parameters()

    def regularize_gradient_norm(self, calculate_loss):
        x = Variable(self.x, requires_grad=True).cuda()
        d1_logits = self.discriminator(x, context={"c":self.c})

After Change

        if self.config.c_discriminator:
            for param in self.c_discriminator.parameters():
                yield param
        if self.config.video_discriminator:
            for param in self.video_discriminator.parameters():
                yield param
        if self.config.image_discriminator:
            for param in self.image_discriminator.parameters():
                yield param

Instances