diff options
author | 2018-04-13 05:47:21 -0700 | |
---|---|---|
committer | 2018-04-13 05:49:43 -0700 | |
commit | 345414cb4fa43af5906adc64a380986eaade4f53 (patch) | |
tree | 5dae2dc6ffe76f063ccc9713e3acec8a1c671652 /tensorflow/contrib/kfac | |
parent | b520022c95b246749fa3f63ca818058c22944720 (diff) |
- Fixed small bug in example script
PiperOrigin-RevId: 192756152
Diffstat (limited to 'tensorflow/contrib/kfac')
-rw-r--r-- | tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py b/tensorflow/contrib/kfac/examples/convnet_mnist_single_main.py index 3aa52aff19..2c1f099360 100644 --- 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__": |