Skip to content

GitLab

Commit 801c522d authored by Zongwei Zhou's avatar Zongwei Zhou Committed by A. Unique TensorFlower
Browse files

Correct Resnet CTL data dir 

PiperOrigin-RevId: 299764679
parent 1212599e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ 801c522d
...@@ -392,8 +392,7 @@ class Resnet50CtlBenchmarkReal(Resnet50CtlBenchmarkBase): ...@@ -392,8 +392,7 @@ class Resnet50CtlBenchmarkReal(Resnet50CtlBenchmarkBase):
def __init__(self, output_dir=None, root_data_dir=None, **kwargs): def __init__(self, output_dir=None, root_data_dir=None, **kwargs):
def_flags = {} def_flags = {}
def_flags['skip_eval'] = True def_flags['skip_eval'] = True
def_flags['data_dir'] = ('/readahead/200M/placer/prod/home/distbelief/' def_flags['data_dir'] = os.path.join(root_data_dir, 'imagenet')
'imagenet-tensorflow/imagenet-2012-tfrecord')
def_flags['train_steps'] = 110 def_flags['train_steps'] = 110
def_flags['steps_per_loop'] = 20 def_flags['steps_per_loop'] = 20
def_flags['log_steps'] = 10 def_flags['log_steps'] = 10
......
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