Skip to content

GitLab

Unverified Commit 00b081e9 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

points default data_dir to child folder. (#6131) 

Failed test is python2  and was a kokoro failure
parent 44ecc4a7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/cifar10_main.py
View file @ 00b081e9
...@@ -234,7 +234,7 @@ def cifar10_model_fn(features, labels, mode, params): ...@@ -234,7 +234,7 @@ def cifar10_model_fn(features, labels, mode, params):
def define_cifar_flags(): def define_cifar_flags():
resnet_run_loop.define_resnet_flags() resnet_run_loop.define_resnet_flags()
flags.adopt_module_key_flags(resnet_run_loop) 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', model_dir='/tmp/cifar10_model',
resnet_size='56', resnet_size='56',
train_epochs=182, train_epochs=182,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment