feat: remove print statement

patil-suraj · Mar 24, 2022 · f49a07f · f49a07f
1 parent 11b53a4
commit f49a07f
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/vqgan_jax/modeling_flax_vqgan.py b/vqgan_jax/modeling_flax_vqgan.py
@@ -399,8 +399,6 @@ def setup(self):
                                                     - 1]
     curr_res = self.config.resolution // 2**(self.config.num_resolutions - 1)
     self.z_shape = (1, self.config.z_channels, curr_res, curr_res)
-    print("Working with z of shape {} = {} dimensions.".format(
-        self.z_shape, np.prod(self.z_shape)))
 
     # z to block_in
     self.conv_in = nn.Conv(