part_generator.py [685:697]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            style = get_latents_fn(batch_size, num_layers, latent_dim)
            noise = image_noise(batch_size, image_size)

            bitmap_feats = self.GAN.Enc(image_cond_batch)

            w_space = latent_to_w(self.GAN.S, style)
            w_styles = styles_def_to_tensor(w_space)

            generated_partial_images = self.GAN.G(w_styles, noise, bitmap_feats)
            generated_images = torch.max(generated_partial_images, image_partial_batch)

            generated_image_stack_batch = torch.cat([image_cond_batch[:, :self.partid], torch.max(generated_partial_images, image_cond_batch[:, self.partid:self.partid+1]),
                                                    image_cond_batch[:, self.partid+1:-1], generated_images], 1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



part_generator.py [729:741]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            style = get_latents_fn(batch_size, num_layers, latent_dim)
            noise = image_noise(batch_size, image_size)

            bitmap_feats = self.GAN.Enc(image_cond_batch)

            w_space = latent_to_w(self.GAN.S, style)
            w_styles = styles_def_to_tensor(w_space)

            generated_partial_images = self.GAN.G(w_styles, noise, bitmap_feats)
            generated_images = torch.max(generated_partial_images, image_partial_batch)
            
            generated_image_stack_batch = torch.cat([image_cond_batch[:, :self.partid], torch.max(generated_partial_images, image_cond_batch[:, self.partid:self.partid+1]),
                                                    image_cond_batch[:, self.partid+1:-1], generated_images], 1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -