Skip to content

Commit 499e601

Browse files
authoredSep 16, 2024
adding customized warmup steps (#2451)
1 parent 44f3c8c commit 499e601

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)
Please sign in to comment.