Skip to content

GitLab

Commit 158c96d2 authored by Sai Ganesh Bandiatmakuri's avatar Sai Ganesh Bandiatmakuri Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 309761595
parent ef7b1dac
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/benchmark/perfzero_benchmark.py
View file @ 158c96d2
...@@ -72,6 +72,8 @@ class PerfZeroBenchmark(tf.test.Benchmark): ...@@ -72,6 +72,8 @@ class PerfZeroBenchmark(tf.test.Benchmark):
# TPU models are expected to accept a --tpu=name flag. PerfZero creates # TPU models are expected to accept a --tpu=name flag. PerfZero creates
# the TPU at runtime and passes the TPU's name to this flag. # the TPU at runtime and passes the TPU's name to this flag.
self.default_flags['tpu'] = resolved_tpu self.default_flags['tpu'] = resolved_tpu
else:
self.default_flags['tpu'] = ''
logging.info('root_data_dir: %s', root_data_dir) logging.info('root_data_dir: %s', root_data_dir)
......
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