change: fix horovod mnist script (#224)

Richardwan7 · laurenyu · commit 88b06d2bccb1 · 2019-07-19T17:42:40.000-07:00
diff --git a/test/resources/mnist/horovod_mnist.py b/test/resources/mnist/horovod_mnist.py
@@ -117,7 +117,7 @@
 
 if hvd.rank() == 0:
     # Exports the keras model as TensorFlow Serving Saved Model
-    with tf.Session() as session:
+    with K.get_session() as session:
 
         init = tf.global_variables_initializer()
         session.run(init)
