Skip to content

GitLab

Commit f23d7bc4 authored by Sergey Kozub's avatar Sergey Kozub Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 481073259
parent 721a6fa4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/shakespeare_benchmark.py
View file @ f23d7bc4
...@@ -331,7 +331,7 @@ class ShakespeareKerasBenchmarkReal(ShakespeareBenchmarkBase): ...@@ -331,7 +331,7 @@ class ShakespeareKerasBenchmarkReal(ShakespeareBenchmarkBase):
def benchmark_xla_8_gpu(self): def benchmark_xla_8_gpu(self):
"""Benchmark 8 gpu w/xla.""" """Benchmark 8 gpu w/xla."""
self._setup() self._setup()
FLAGS.num_gpus = 1 FLAGS.num_gpus = 8
FLAGS.batch_size = 64 * 8 FLAGS.batch_size = 64 * 8
FLAGS.log_steps = 10 FLAGS.log_steps = 10
FLAGS.enable_xla = True FLAGS.enable_xla = 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