points default data_dir to child folder. (#6131)

Failed test is python2 and was a kokoro failure

points default data_dir to child folder. (#6131)
Failed test is python2 and was a kokoro failure
00b081e9 · Toby Boyd · GitHub · 44ecc4a7 · 00b081e9
Unverified Commit 00b081e9 authored Feb 01, 2019 by Toby Boyd Committed by GitHub Feb 01, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/resnet/cifar10_main.py official/resnet/cifar10_main.py +1 -1

No files found.
--- a/official/resnet/cifar10_main.py
+++ b/official/resnet/cifar10_main.py
@@ -234,7 +234,7 @@ def cifar10_model_fn(features, labels, mode, params):
 def define_cifar_flags():
  resnet_run_loop.define_resnet_flags()
  flags.adopt_module_key_flags(resnet_run_loop)
-  flags_core.set_defaults(data_dir='/tmp/cifar10_data',
+  flags_core.set_defaults(data_dir='/tmp/cifar10_data/cifar-10-batches-bin',
                          model_dir='/tmp/cifar10_model',
                          resnet_size='56',
                          train_epochs=182,