fix bug with skip layer excitation

lucidrains · Nov 16, 2020 · 51fc0ee · 51fc0ee
1 parent dc348de
commit 51fc0ee
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
@@ -306,7 +306,7 @@ def forward(self, x):
                 residuals[out_res] = residual
 
             if res in residuals:
-                x = x + residuals[res]
+                x = x * residuals[res]
 
         x = self.out_conv(x)
         return x.tanh()

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