Skip to content

Commit 684a161

Browse files
authored
After-the-fact response to @k-w-w 's review (#3841)
1 parent 59c7683 commit 684a161

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

research/tensorrt/tensorrt.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -329,10 +329,6 @@ def log_stats(graph_name, log_buffer, timings, batch_size):
329329
timings: list of floats, times produced for multiple runs that will be
330330
used for statistic calculation
331331
batch_size: int, number of examples per batch
332-
333-
Returns:
334-
list of two floats representing frames per second and 99th percentile
335-
time per batch
336332
"""
337333
times = np.array(timings)
338334
steps = len(times)
@@ -357,8 +353,6 @@ def log_stats(graph_name, log_buffer, timings, batch_size):
357353

358354
log_buffer.write(msg)
359355

360-
return [speed_mean, time_99th]
361-
362356

363357
def time_and_log_graph(graph_name, graph_def, data, log_buffer, flags):
364358
timings, result = time_graph(

0 commit comments

Comments
 (0)