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
Hide 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):
def benchmark_xla_8_gpu(self):
"""Benchmark 8 gpu w/xla."""
self._setup()
FLAGS.num_gpus = 1
FLAGS.num_gpus = 8
FLAGS.batch_size = 64 * 8
FLAGS.log_steps = 10
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