提交 47fda4a3 编写于 作者: I Ihor Indyk 提交者: TensorFlower Gardener

[tf.data] Increase the number of `iterator.get_next()` calls to 10000 (from...

[tf.data] Increase the number of `iterator.get_next()` calls to 10000 (from 1000) in `map_and_interleave` autotuning benchmark to decrease variance of the output.

PiperOrigin-RevId: 257624991
上级 3e799d06
......@@ -189,14 +189,14 @@ class AutotuneBenchmark(test.Benchmark):
with session.Session() as sess:
for _ in range(5):
sess.run(get_next)
for _ in range(1000):
for _ in range(10000):
start = time.time()
sess.run(get_next)
end = time.time()
deltas.append(end - start)
self.report_benchmark(
iters=1000,
iters=10000,
wall_time=np.median(deltas),
name="map_and_interleave" + ("_autotune" if autotune else ""))
return np.median(deltas)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册