update to test in servers
This commit is contained in:
@@ -68,6 +68,7 @@ def clamped_act(z):
|
||||
|
||||
@jit
|
||||
def inv_act(z):
|
||||
z = jnp.maximum(z, 1e-7)
|
||||
return 1 / z
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user