remove tanh nonlinearity

lucidrains · Jan 19, 2021 · ddcf3d7 · ddcf3d7
1 parent 0d84085
commit ddcf3d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deep_daze/deep_daze.py b/deep_daze/deep_daze.py
@@ -51,7 +51,7 @@ def rand_cutout(image, size):
 # load siren
 
 def norm_siren_output(img):
-    return (img.tanh() + 1) * 0.5
+    return (img + 1) * 0.5
 
 class DeepDaze(nn.Module):
     def __init__(