Correct Resnet CTL data dir

PiperOrigin-RevId: 299764679

Correct Resnet CTL data dir
PiperOrigin-RevId: 299764679
801c522d · Zongwei Zhou · A. Unique TensorFlower · 1212599e · 801c522d
Commit 801c522d authored Mar 08, 2020 by Zongwei Zhou Committed by A. Unique TensorFlower Mar 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

official/benchmark/resnet_ctl_imagenet_benchmark.py official/benchmark/resnet_ctl_imagenet_benchmark.py +1 -2

No files found.
--- a/official/benchmark/resnet_ctl_imagenet_benchmark.py
+++ b/official/benchmark/resnet_ctl_imagenet_benchmark.py
@@ -392,8 +392,7 @@ class Resnet50CtlBenchmarkReal(Resnet50CtlBenchmarkBase):
  def __init__(self, output_dir=None, root_data_dir=None, **kwargs):
    def_flags = {}
    def_flags['skip_eval'] = True
-    def_flags['data_dir'] = ('/readahead/200M/placer/prod/home/distbelief/'
-                             'imagenet-tensorflow/imagenet-2012-tfrecord')
+    def_flags['data_dir'] = os.path.join(root_data_dir, 'imagenet')
    def_flags['train_steps'] = 110
    def_flags['steps_per_loop'] = 20
    def_flags['log_steps'] = 10