print more shapes rolux#2

morzh · Jul 20, 2020 · 4f8d0b3 · 4f8d0b3
1 parent a5923fe
commit 4f8d0b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/projector.py b/projector.py
@@ -100,6 +100,7 @@ def set_network(self, Gs, minibatch_size=1):
         print('dlatents_noise shape', dlatents_noise.shape)
         print('dlatents_var shape', self._dlatents_var.shape)
         self._dlatents_expr = self._dlatents_var + dlatents_noise
+        print('_dlatents_expr computed')
         self._images_expr = self._Gs.components.synthesis.get_output_for(self._dlatents_expr, randomize_noise=False)
 
         # Downsample image to 256x256 if it's larger than that. VGG was built for 224x224 images.