initialize methods

2024-05-21 14:34:01 +08:00
parent 0e89ed1d7c
commit 40b7d8360c
46 changed files with 222 additions and 40 deletions
--- a/tensorneat/utils/activation.py
+++ b/tensorneat/utils/activation.py
@@ -11,7 +11,7 @@ class Act:

    @staticmethod
    def tanh(z):
-        return jnp.tanh(z)
+        return jnp.tanh(0.6 * z)

    @staticmethod
    def sin(z):