From 812259f442d519531d8506a4bcd7e580e9d86b67 Mon Sep 17 00:00:00 2001
From: Dillon Erb <585865+dte@users.noreply.github.com>
Date: Mon, 27 Jan 2020 15:25:16 -0500
Subject: [PATCH] Update mnist_keras.py

---
 mnist_keras.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mnist_keras.py b/mnist_keras.py
index 9fbbb70..6a33f00 100644
--- a/mnist_keras.py
+++ b/mnist_keras.py
@@ -18,8 +18,8 @@
 
 # Horovod: pin GPU to be used to process local rank (one GPU per process)
 config = tf.ConfigProto()
-config.gpu_options.allow_growth = True
-config.gpu_options.visible_device_list = str(hvd.local_rank())
+# config.gpu_options.allow_growth = True
+# config.gpu_options.visible_device_list = str(hvd.local_rank())
 K.set_session(tf.Session(config=config))
 
 batch_size = 128