fix a bug where style and noise was being refetched in generator trai…

…ning
lucidrains · Feb 18, 2020 · 0b061d4 · 0b061d4
1 parent 8e4ebf8
commit 0b061d4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
   name = 'stylegan2_pytorch',
   packages = find_packages(),
   scripts=['bin/stylegan2_pytorch'],
-  version = '0.4.8',
+  version = '0.4.9',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',

diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py
@@ -511,8 +511,6 @@ def train(self):
         self.GAN.G_opt.zero_grad()
         for i in range(self.gradient_accumulate_every):
             style, noise = inputs[i]
-            style = get_latents_fn(batch_size, num_layers, latent_dim)
-            noise = image_noise(batch_size, image_size)
 
             w_space = latent_to_w(self.GAN.S, style)
             w_styles = styles_def_to_tensor(w_space)