Adjust number of warmup loops (#4005)

5be198ec · Karmel Allison · GitHub · f7f2bf4f · 5be198ec
Unverified Commit 5be198ec authored Apr 17, 2018 by Karmel Allison Committed by GitHub Apr 17, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/tensorrt/tensorrt.py research/tensorrt/tensorrt.py +1 -1

No files found.
--- a/research/tensorrt/tensorrt.py
+++ b/research/tensorrt/tensorrt.py
@@ -36,7 +36,7 @@ import tensorflow.contrib.tensorrt as trt
 from official.resnet import imagenet_preprocessing  # pylint: disable=g-bad-import-order

 _GPU_MEM_FRACTION = 0.50
-_WARMUP_NUM_LOOPS = 50
+_WARMUP_NUM_LOOPS = 5
 _LOG_FILE = "log.txt"
 _LABELS_FILE = "labellist.json"
 _GRAPH_FILE = "frozen_graph.pb"