[WideAndDeep/TF] scripts fix

Author: nv-kkudrynski

TensorFlow/Recommendation/WideAndDeep/scripts/DGX1_benchmark_training_amp_1gpu.sh

@@ -18,8 +18,8 @@ set -x
 set -e
 
 python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark \
   --amp

TensorFlow/Recommendation/WideAndDeep/scripts/DGX1_benchmark_training_amp_8gpu.sh

@@ -19,8 +19,8 @@ set -e
 
 mpiexec --allow-run-as-root --bind-to socket -np 8 \
   python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark \
   --amp \

TensorFlow/Recommendation/WideAndDeep/scripts/DGX1_benchmark_training_fp32_1gpu.sh

@@ -18,7 +18,7 @@ set -x
 set -e
 
 python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark

TensorFlow/Recommendation/WideAndDeep/scripts/DGX1_benchmark_training_fp32_8gpu.sh

@@ -20,7 +20,7 @@ set -e
 mpiexec --allow-run-as-root --bind-to socket -np 8 \
   python -m trainer.task \
   --hvd \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark

TensorFlow/Recommendation/WideAndDeep/scripts/DGXA100_benchmark_training_amp_1gpu.sh

@@ -18,8 +18,8 @@ set -x
 set -e
 
 python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark \
   --amp

TensorFlow/Recommendation/WideAndDeep/scripts/DGXA100_benchmark_training_amp_8gpu.sh

@@ -19,8 +19,8 @@ set -e
 
 mpiexec --allow-run-as-root --bind-to socket -np 8 \
   python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark \
   --amp \

TensorFlow/Recommendation/WideAndDeep/scripts/DGXA100_benchmark_training_tf32_1gpu.sh

@@ -18,7 +18,7 @@ set -x
 set -e
 
 python -m trainer.task \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark

TensorFlow/Recommendation/WideAndDeep/scripts/DGXA100_benchmark_training_tf32_8gpu.sh

@@ -20,7 +20,7 @@ set -e
 mpiexec --allow-run-as-root --bind-to socket -np 8 \
   python -m trainer.task \
   --hvd \
-  --benchmark_warmup_steps 50 \
-  --benchmark_steps 200 \
+  --benchmark_warmup_steps 500 \
+  --benchmark_steps 1000 \
   --gpu \
   --benchmark