From c52b6e61769e4f8f03f20325a1dbdd30024c74ce Mon Sep 17 00:00:00 2001
From: Ashiq Imran <ashiq.imran@intel.com>
Date: Mon, 16 Sep 2024 12:22:14 -0700
Subject: [PATCH] adding customized warmup steps (#2451)

---
 benchmarks/common/tensorflow/start.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmarks/common/tensorflow/start.sh b/benchmarks/common/tensorflow/start.sh
index 8b2f2fac8..06716d566 100644
--- a/benchmarks/common/tensorflow/start.sh
+++ b/benchmarks/common/tensorflow/start.sh
@@ -2021,6 +2021,7 @@ function graphsage() {
         fi
 
         CMD="${CMD} $(add_arg "--pretrained-model" ${PRETRAINED_MODEL})"
+	CMD="${CMD} $(add_arg "--warmup-steps" ${WARMUP_STEPS})"
         CMD="${CMD} $(add_arg "--steps" ${STEPS})"
         CMD=${CMD} run_model
       else