Change to dense layer (#270)

mlcommons · May 7, 2019 · 6cb3d42 · 6cb3d42
1 parent 1c6ae72
commit 6cb3d42
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/image_classification/tensorflow/official/resnet/resnet_model.py b/image_classification/tensorflow/official/resnet/resnet_model.py
@@ -453,7 +453,10 @@ def __call__(self, inputs, training):
       inputs = tf.reshape(inputs, [-1, self.final_size])
       mlperf_log.resnet_print(key=mlperf_log.MODEL_HP_DENSE,
                               value=self.num_classes)
-      inputs = tf.layers.dense(inputs=inputs, units=self.num_classes)
+      inputs = tf.layers.dense(
+        inputs=inputs,
+        units=self.num_classes,
+        kernel_initializer=tf.random_normal_initializer(stddev=.01))
       inputs = tf.identity(inputs, 'final_dense')
 
       # Drop batch size from shape logging.