Unverified Commit 68b724ce authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Fix lint issues (#6204)

parent 5d3a7d04
...@@ -106,6 +106,7 @@ class EstimatorCifar10BenchmarkTests(tf.test.Benchmark): ...@@ -106,6 +106,7 @@ class EstimatorCifar10BenchmarkTests(tf.test.Benchmark):
self._run_and_report_benchmark() self._run_and_report_benchmark()
def _run_and_report_benchmark(self): def _run_and_report_benchmark(self):
"""Executes benchmark and reports result."""
start_time_sec = time.time() start_time_sec = time.time()
stats = cifar_main.run_cifar(flags.FLAGS) stats = cifar_main.run_cifar(flags.FLAGS)
wall_time_sec = time.time() - start_time_sec wall_time_sec = time.time() - start_time_sec
......
...@@ -67,6 +67,7 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark): ...@@ -67,6 +67,7 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark):
self._run_and_report_benchmark() self._run_and_report_benchmark()
def benchmark_8_gpu_bfc_allocator(self): def benchmark_8_gpu_bfc_allocator(self):
"""Restricts CPU memory allocation."""
self._setup() self._setup()
FLAGS.num_gpus = 8 FLAGS.num_gpus = 8
FLAGS.enable_eager = True FLAGS.enable_eager = True
......
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