Temporarily make the SBSA short-run shorter due to slow CI runner (#5524

) Signed-off-by: Kamil Tokarski <[email protected]>
NVIDIA · Jun 14, 2024 · 1238d5d · 1238d5d
1 parent eb34a22
commit 1238d5d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/qa/TL3_RN50_short/test_pytorch.sh b/qa/TL3_RN50_short/test_pytorch.sh
@@ -22,7 +22,7 @@ fi
 LOG=dali.log
 
 SECONDS=0
-torchrun --nproc_per_node=${NUM_GPUS} main.py -a resnet50 --b 256 --loss-scale 128.0 --workers 8 --lr=0.4 --fp16-mode --epochs 2 ./ 2>&1 | tee $LOG
+torchrun --nproc_per_node=${NUM_GPUS} main.py -a resnet50 --b 256 --loss-scale 128.0 --workers 8 --lr=0.4 --fp16-mode --epochs 1 ./ 2>&1 | tee $LOG
 
 RET=${PIPESTATUS[0]}
 echo "Training ran in $SECONDS seconds"
@@ -31,9 +31,9 @@ if [[ $RET -ne 0 ]]; then
     CLEAN_AND_EXIT 2
 fi
 
-MIN_TOP1=20.0
-MIN_TOP5=40.0
-MIN_PERF=100
+MIN_TOP1=10.0
+MIN_TOP5=20.0
+MIN_PERF=50
 
 TOP1=$(grep "^##Top-1" $LOG | awk '{print $2}')
 TOP5=$(grep "^##Top-5" $LOG | awk '{print $2}')