tensorflow / models

Models and examples built with TensorFlow
Other
76.99k stars 45.79k forks source link

Update freezable_sync_batch_norm.py #11209

Open ZoeyZiyunWang opened 4 months ago

ZoeyZiyunWang commented 4 months ago

This code is not compatible with the newest version of Tensorflow (2.16.1). Therefore I Replace the line: "class FreezableSyncBatchNorm(tf.keras.layers.experimental.SyncBatchNormalization):" With: "class FreezableSyncBatchNorm(tf.keras.layers.BatchNormalization):" to make it compatible with the newest Tenserflow version (2.16.1).

Description

:memo: Please include a summary of the change.

  • Please also include relevant motivation and context.
  • List any dependencies that are required for this change.

Type of change

For a new feature or function, please create an issue first to discuss it with us before submitting a pull request.

Note: Please delete options that are not relevant.

Tests

:memo: Please describe the tests that you ran to verify your changes.

  • Provide instructions so we can reproduce.
  • Please also list any relevant details for your test configuration.

Test Configuration:

Checklist

google-cla[bot] commented 4 months ago

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.