google-research

1
--- external/stylegan/training/networks_stylegan3_sky.py	2023-03-09 18:10:40.036892930 +0000
2
+++ external_reference/stylegan/training/networks_stylegan3_sky.py	2023-03-09 18:07:58.562065004 +0000
3
@@ -642,6 +642,7 @@
4
         z_dim,                      # Input latent (Z) dimensionality.
5
         c_dim,                      # Conditioning label (C) dimensionality.
6
         w_dim,                      # Intermediate latent (W) dimensionality.
7
+        enc_dim, # encoder dimension
8
         img_resolution,             # Output resolution.
9
         img_channels,               # Number of output color channels.
10
         mapping_kwargs       = {},  # Arguments for MappingNetwork.
11
@@ -653,12 +654,13 @@
12
         self.z_dim = z_dim
13
         self.c_dim = c_dim
14
         self.w_dim = w_dim
15
+        self.enc_dim = enc_dim
16
         self.img_resolution = img_resolution
17
         self.img_channels = img_channels
18
         self.training_mode = training_mode
19
         self.scale_mapping_kwargs = scale_mapping_kwargs
20
         use_scale_affine = True if 'patch' in self.training_mode else False # add affine layer on style input
21
-        self.synthesis = SynthesisNetwork(w_dim=w_dim, img_resolution=img_resolution, img_channels=img_channels,
22
+        self.synthesis = SynthesisNetwork(w_dim=w_dim+enc_dim, img_resolution=img_resolution, img_channels=img_channels,
23
                                           training_mode=training_mode, use_scale_affine=use_scale_affine,
24
                                           **synthesis_kwargs)
25
         self.num_ws = self.synthesis.num_ws
26