From 748ef1c2eb58b0e031d796ea0211a8c6d74531ff Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 18 Jul 2018 09:55:53 -0700 Subject: Add back in stddev parameter inadvertently dropped from conv model. Use create_eval_graph() for graph to be frozen. PiperOrigin-RevId: 205093459 --- tensorflow/examples/speech_commands/freeze.py | 2 +- tensorflow/examples/speech_commands/models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'tensorflow/examples') diff --git a/tensorflow/examples/speech_commands/freeze.py b/tensorflow/examples/speech_commands/freeze.py index 7657b23c60..89e790d4e4 100644 --- a/tensorflow/examples/speech_commands/freeze.py +++ b/tensorflow/examples/speech_commands/freeze.py @@ -130,7 +130,7 @@ def main(_): FLAGS.clip_stride_ms, FLAGS.window_size_ms, FLAGS.window_stride_ms, FLAGS.feature_bin_count, FLAGS.model_architecture, FLAGS.preprocess) if FLAGS.quantize: - tf.contrib.quantize.create_training_graph(quant_delay=0) + tf.contrib.quantize.create_eval_graph() models.load_variables_from_checkpoint(sess, FLAGS.start_checkpoint) # Turn all the variables into inline constants inside the graph and save it. diff --git a/tensorflow/examples/speech_commands/models.py b/tensorflow/examples/speech_commands/models.py index 65ae3b1511..4d1454be0d 100644 --- a/tensorflow/examples/speech_commands/models.py +++ b/tensorflow/examples/speech_commands/models.py @@ -302,7 +302,7 @@ def create_conv_model(fingerprint_input, model_settings, is_training): label_count = model_settings['label_count'] final_fc_weights = tf.get_variable( name='final_fc_weights', - initializer=tf.truncated_normal_initializer, + initializer=tf.truncated_normal_initializer(stddev=0.01), shape=[second_conv_element_count, label_count]) final_fc_bias = tf.get_variable( name='final_fc_bias', -- cgit v1.2.3