You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm trying to implement a projected gradient descent method and need to apply a constraint to parameters of a nn.Module. For example, apply jnp.abs() to a specific parameter. I haven't been able to find a simple way to do this yet, let me know the there's a canonical approach.
The text was updated successfully, but these errors were encountered:
I'm trying to implement a projected gradient descent method and need to apply a constraint to parameters of a
nn.Module
. For example, applyjnp.abs()
to a specific parameter. I haven't been able to find a simple way to do this yet, let me know the there's a canonical approach.The text was updated successfully, but these errors were encountered: