5f65493b550ab85af30666e60e27248ae3df7b00,hypergan/gans/aligned_interpolate_gan.py,AlignedInterpolateGAN,create,#AlignedInterpolateGAN#,29

Before Change

    def create(self):
        self.latent = self.create_component("latent")
        self.x = self.inputs.next()[0]
        self.generator_train = self.create_component("generator", input=self.x, context_shapes={"gamma": [1]})
        self.discriminator = self.create_component("discriminator", context_shapes={"index": [1]})
        self.loss = self.create_component("loss")
        self.trainer = self.create_component("trainer")

After Change

        if self.config.shared_discriminator:
            self.discriminator = self.create_component("discriminator", context_shapes={"class": [1]})
        else:
            self.discriminator = self.create_component("discriminator")
            self.discriminator2 = self.create_component("discriminator")
        self.loss = self.create_component("loss")
        self.trainer = self.create_component("trainer")
        self.sigmoid = torch.nn.Sigmoid()

Instances