fix bug with discriminator attention, thanks to @ferrophile

lucidrains · Mar 2, 2021 · 24098fa · 24098fa
1 parent 54eb6d9
commit 24098fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
@@ -567,7 +567,7 @@ def __init__(
         self.residual_layers = nn.ModuleList([])
 
         for (res, ((_, chan_in), (_, chan_out))) in zip(non_residual_resolutions, chan_in_out):
-            image_width = 2 ** resolution
+            image_width = 2 ** res
 
             attn = None
             if image_width in attn_res_layers:

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