Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Switch to default tf.keras.layers.BatchNormalization #188

Merged
merged 1 commit into from
Apr 30, 2020

Commits on Feb 25, 2020

  1. Switch to default tf.keras.layers.BatchNormalization

    As stated in the official tf documentation, now setting trainable=False will run the layer in inference mode, so when the layer is frozen it will use moving mean and moving variance instead of current batch mean and variance:
    https://www.tensorflow.org/api_docs/python/tf/keras/layers/BatchNormalization
    
    So, there is no need to use a custom BatchNormalization layer anymore.
    Maxinho96 committed Feb 25, 2020
    Configuration menu
    Copy the full SHA
    4bf1dd9 View commit details
    Browse the repository at this point in the history