adding customized warmup steps (#2451)

intel · Sep 16, 2024 · 499e601 · 499e601
1 parent 44f3c8c
commit 499e601
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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