update to test in servers

This commit is contained in:
wls2002
2023-05-10 22:33:51 +08:00
parent ce35b01896
commit b271a56827
9 changed files with 112 additions and 34 deletions

View File

@@ -68,6 +68,7 @@ def clamped_act(z):
@jit
def inv_act(z):
z = jnp.maximum(z, 1e-7)
return 1 / z