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
-
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