adding customized warmup steps (#2451)

intel · Sep 16, 2024 · c52b6e6 · c52b6e6
1 parent 594fd1d
commit c52b6e6
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