From 63ff2d043e72685f10e862a9dd0b1775640e2fce Mon Sep 17 00:00:00 2001 From: rifhath162 <147535738+rifhath162@users.noreply.github.com> Date: Tue, 11 Jun 2024 23:46:51 +0100 Subject: [PATCH] Update freezable_sync_batch_norm.py --- research/object_detection/core/freezable_sync_batch_norm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/research/object_detection/core/freezable_sync_batch_norm.py b/research/object_detection/core/freezable_sync_batch_norm.py index f95a1064983..53ef8ac9026 100644 --- a/research/object_detection/core/freezable_sync_batch_norm.py +++ b/research/object_detection/core/freezable_sync_batch_norm.py @@ -17,8 +17,8 @@ import tensorflow as tf -class FreezableSyncBatchNorm(tf.keras.layers.experimental.SyncBatchNormalization - ): +class FreezableSyncBatchNorm(tf.keras.layers.SyncBatchNormalization): + """Sync Batch normalization layer (Ioffe and Szegedy, 2014). This is a `freezable` batch norm layer that supports setting the `training`