fix pl_mean initialized value

lucidrains · Jul 3, 2020 · 611ab46 · 611ab46
1 parent 8ef4b20
commit 611ab46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 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.17.5',
+  version = '0.17.6',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',

diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py
@@ -629,7 +629,7 @@ def __init__(self, name, results_dir, models_dir, image_size, network_capacity,
         self.av = None
         self.trunc_psi = trunc_psi
 
-        self.pl_mean = 0
+        self.pl_mean = None
 
         self.gradient_accumulate_every = gradient_accumulate_every