Internal change

PiperOrigin-RevId: 394577222

Internal change
PiperOrigin-RevId: 394577222
8fcf177e · Isha Arkatkar · A. Unique TensorFlower · 2cd6f1c6 · 8fcf177e
Commit 8fcf177e authored Sep 02, 2021 by Isha Arkatkar Committed by A. Unique TensorFlower Sep 02, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

official/vision/image_classification/resnet/resnet_model.py official/vision/image_classification/resnet/resnet_model.py +1 -2

No files found.
--- a/official/vision/image_classification/resnet/resnet_model.py
+++ b/official/vision/image_classification/resnet/resnet_model.py
@@ -311,8 +311,7 @@ def resnet50(num_classes,
  x = layers.GlobalAveragePooling2D()(x)
  x = layers.Dense(
      num_classes,
-      kernel_initializer=tf.compat.v1.keras.initializers.random_normal(
-          stddev=0.01),
+      kernel_initializer=tf.initializers.random_normal(stddev=0.01),
      kernel_regularizer=_gen_l2_regularizer(use_l2_regularizer),
      bias_regularizer=_gen_l2_regularizer(use_l2_regularizer),
      name='fc1000')(