From ddcf3d77132686418a1c68b8f50381adb373d5a5 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Mon, 18 Jan 2021 20:50:21 -0800
Subject: [PATCH] remove tanh nonlinearity

---
 deep_daze/deep_daze.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deep_daze/deep_daze.py b/deep_daze/deep_daze.py
index 6f4fe1c..b04e3d5 100644
--- 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__(