add save function in pipeline

2024-06-16 21:47:53 +08:00
parent b9d6482d11
commit fb2ae5d2fa
10 changed files with 94 additions and 164 deletions
--- a/tensorneat/utils/activation/act_jnp.py
+++ b/tensorneat/utils/activation/act_jnp.py
@@ -14,7 +14,8 @@ class Act:

    @staticmethod
    def tanh(z):
-        return jnp.tanh(0.6 * z)
+        z = jnp.clip(0.6*z, -3, 3)
+        return jnp.tanh(z)

    @staticmethod
    def sin(z):