Is there synchronized BatchNormalization for JAX or torch backends? #20354
caiuspetronius
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Currently, Keras only accepts synchronized = True in the BatchNormalization() while using the TF backend. How do I achieve the same with the JAX or torch backend?
Beta Was this translation helpful? Give feedback.
All reactions