remove sigmoid gating

mtlong · Apr 9, 2021 · 6be0c37 · 6be0c37
1 parent 97e5cac
commit 6be0c37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'siren-pytorch',
   packages = find_packages(),
-  version = '0.1.4',
+  version = '0.1.5',
   license='MIT',
   description = 'Implicit Neural Representations with Periodic Activation Functions',
   author = 'Phil Wang',

diff --git a/siren_pytorch/siren_pytorch.py b/siren_pytorch/siren_pytorch.py
@@ -83,7 +83,7 @@ def forward(self, x, mods = None):
             x = layer(x)
 
             if exists(mod):
-                x *= rearrange(mod, 'd -> () d').sigmoid()
+                x *= rearrange(mod, 'd -> () d')
 
         return self.last_layer(x)