projects / platform / upstream / tensorflow.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: b520022)
- Fixed small bug in example script
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 13 Apr 2018 12:47:21 +0000 (05:47 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 13 Apr 2018 12:49:43 +0000 (05:49 -0700)
PiperOrigin-RevId: 192756152

tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py patch | blob | history

diff --git a/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py b/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py
index 3aa52af..2c1f099 100644 (file)
--- a/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py
+++ b/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py
@@ -32,7 +32,7 @@ flags.DEFINE_string("data_dir", "/tmp/mnist", "local mnist dir")
 
 
 def main(unused_argv):
-  convnet.train_mnist_single_gpu(FLAGS.data_dir, num_epochs=200)
+  convnet.train_mnist_single_machine(FLAGS.data_dir, num_epochs=200)
 
 
 if __name__ == "__main__":
Domain: Machine Learning / ML Framework;