Skip to content

GitLab

Commit 225eda71 authored by Will Cromar's avatar Will Cromar Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 300850465
parent c25e3e01
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ 225eda71
...@@ -87,10 +87,9 @@ class CtlBenchmark(PerfZeroBenchmark): ...@@ -87,10 +87,9 @@ class CtlBenchmark(PerfZeroBenchmark):
# first entry in the time_log is start of step 0. The rest of the # first entry in the time_log is start of step 0. The rest of the
# entries are the end of each step recorded # entries are the end of each step recorded
time_log = stats['step_timestamp_log'] time_log = stats['step_timestamp_log']
elapsed = time_log[-1].timestamp - time_log[warmup].timestamp steps_elapsed = time_log[-1].batch_index - time_log[warmup].batch_index
num_examples = ( time_elapsed = time_log[-1].timestamp - time_log[warmup].timestamp
total_batch_size * log_steps * (len(time_log) - warmup - 1)) examples_per_sec = total_batch_size * (steps_elapsed / time_elapsed)
examples_per_sec = num_examples / elapsed
metrics.append({'name': 'exp_per_second', 'value': examples_per_sec}) metrics.append({'name': 'exp_per_second', 'value': examples_per_sec})
if 'avg_exp_per_second' in stats: if 'avg_exp_per_second' in stats:
......
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