9a9c40b5c0fd5fa11326968a49ba7793c8626bb4,GANs/pix2pixHD/trainer.py,Trainer,train,#Trainer#,51

Before Change

        // generator
        // Note that only global generator would be used in the case of g_scales = 1.
        generator = LocalGenerator()
        fake, _, = generator(x, self.g_n_scales)
        fake.persistent = True
        unlinked_fake = fake.get_unlinked_variable(need_grad=True)

        // discriminator

After Change

        // generator
        // Note that only global generator would be used in the case of g_scales = 1.
        generator = LocalGenerator()
        fake, _, = generator(x,
                             lg_channels=self.model_conf.lg_channels,
                             gg_channels=self.model_conf.gg_channels,
                             n_scales=self.model_conf.g_n_scales,
                             lg_n_residual_layers=self.model_conf.lg_num_residual_loop,
                             gg_n_residual_layers=self.model_conf.gg_num_residual_loop)
        unlinked_fake = fake.get_unlinked_variable(need_grad=True)

        // discriminator
        discriminator = Discriminator()

Instances