Commit 4fa82ae1 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

additional fix for resnet ctl benchamrk. As copybara needs exact spacing in...

additional fix for resnet ctl benchamrk. As copybara needs exact spacing in the file, adding back the missing space.

PiperOrigin-RevId: 270167733
parent f28059c0
...@@ -317,8 +317,7 @@ class Resnet50CtlBenchmarkReal(Resnet50CtlBenchmarkBase): ...@@ -317,8 +317,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['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