fixed a bug in SCIBlock

Meatssauce · Sep 14, 2021 · 7e93637 · 7e93637
1 parent 3a4cacc
commit 7e93637
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/SCINet.py b/SCINet.py
@@ -44,7 +44,7 @@ def call(self, inputs):
         F_odd, F_even = inputs[:, ::2], inputs[:, 1::2]
 
         F_s_odd = F_odd * tf.math.exp(self.conv1ds['phi'](F_even))
-        F_s_even = F_even * tf.math.exp(self.conv1ds['psi'](F_s_odd))
+        F_s_even = F_even * tf.math.exp(self.conv1ds['psi'](F_odd))
 
         F_prime_odd = F_s_odd + self.conv1ds['rho'](F_s_even)
         F_prime_even = F_s_even - self.conv1ds['eta'](F_s_odd)