Skip to content

GitLab

Commit 8ceeb6b3 authored by Isha Arkatkar's avatar Isha Arkatkar Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 394577222
parent 609a5b5e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/vision/image_classification/resnet/resnet_model.py
View file @ 8ceeb6b3
...@@ -311,8 +311,7 @@ def resnet50(num_classes, ...@@ -311,8 +311,7 @@ def resnet50(num_classes,
x = layers.GlobalAveragePooling2D()(x) x = layers.GlobalAveragePooling2D()(x)
x = layers.Dense( x = layers.Dense(
num_classes, num_classes,
kernel_initializer=tf.compat.v1.keras.initializers.random_normal( kernel_initializer=tf.initializers.random_normal(stddev=0.01),
stddev=0.01),
kernel_regularizer=_gen_l2_regularizer(use_l2_regularizer), kernel_regularizer=_gen_l2_regularizer(use_l2_regularizer),
bias_regularizer=_gen_l2_regularizer(use_l2_regularizer), bias_regularizer=_gen_l2_regularizer(use_l2_regularizer),
name='fc1000')( name='fc1000')(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment