Adjust number of warmup loops (tensorflow#4005)

karmel · web-flow · commit 5be198eca885 · 2018-04-17T14:32:19.000-07:00
diff --git a/research/tensorrt/tensorrt.py b/research/tensorrt/tensorrt.py
@@ -36,7 +36,7 @@
 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"
