Skip to content

Commit c52b6e6

Browse files
adding customized warmup steps (#2451)
1 parent 594fd1d commit c52b6e6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

benchmarks/common/tensorflow/start.sh

+1
Original file line numberDiff line numberDiff line change
@@ -2021,6 +2021,7 @@ function graphsage() {
20212021
fi
20222022

20232023
CMD="${CMD} $(add_arg "--pretrained-model" ${PRETRAINED_MODEL})"
2024+
CMD="${CMD} $(add_arg "--warmup-steps" ${WARMUP_STEPS})"
20242025
CMD="${CMD} $(add_arg "--steps" ${STEPS})"
20252026
CMD=${CMD} run_model
20262027
else

0 commit comments

Comments
 (0)