update

dskkato · dskkato · commit b54351e5168e · 2021-07-12T16:36:27.000+09:00
diff --git a/examples/mnist/expected_values.txt b/examples/mnist/expected_values.txt
@@ -1 +1 @@
-9.693555e-06, 3.5453503e-08, 8.619008e-05, 0.000236661, 1.5483033e-07, 7.4597015e-06, 7.556525e-09, 0.9995622, 2.1989117e-06, 9.543356e-05
+1.8487935e-05, 5.188386e-08, 0.0003525585, 0.0014435184, 3.3050244e-07, 3.180701e-05, 2.4361995e-08, 0.9980293, 1.7698758e-05, 0.00010610428
diff --git a/examples/mnist/mnist.py b/examples/mnist/mnist.py
@@ -44,7 +44,7 @@ def probability_model(x):
 
 
 # convert keras model to TF2 function to get a computation graph
-x = tf.TensorSpec((None, 28, 28), tf.float32)
+x = tf.TensorSpec((None, 28, 28), tf.float32, name="x")
 tf_model = probability_model.get_concrete_function(x=x)
 
 # now all variables are converted to constants.
diff --git a/examples/mnist/model.pb b/examples/mnist/model.pb
