Skip to content

[Fix] clipping import#4

Open
RuanJohn wants to merge 1 commit into
evanatyourservice:mainfrom
RuanJohn:main
Open

[Fix] clipping import#4
RuanJohn wants to merge 1 commit into
evanatyourservice:mainfrom
RuanJohn:main

Conversation

@RuanJohn
Copy link
Copy Markdown

clipping is no longer in optax._src. This is a super small fix that just imports clip_by_global_norm directly from optax and removes the unused clipping import from psgd_jax/kron.py.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant