fix bug

lucidrains · Nov 22, 2020 · 4cd81d0 · 4cd81d0
1 parent d95f794
commit 4cd81d0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
@@ -305,7 +305,7 @@ def __init__(
     ):
         super().__init__()
         resolution = log2(image_size)
-        assert is_power_of_two(resolution), 'image size must be a power of 2'
+        assert is_power_of_two(image_size), 'image size must be a power of 2'
         init_channel = 4 if transparent else 3
         fmap_max = default(fmap_max, latent_dim)
 
@@ -444,7 +444,7 @@ def __init__(
     ):
         super().__init__()
         resolution = log2(image_size)
-        assert is_power_of_two(resolution), 'image size must be a power of 2'
+        assert is_power_of_two(image_size), 'image size must be a power of 2'
         assert disc_output_size in {1, 5}, 'discriminator output dimensions can only be 5x5 or 1x1'
 
         resolution = int(resolution)

diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.9.1'
+__version__ = '0.9.2'