Commit1238d5d

committed

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

Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>

1 parenteb34a22 commit1238d5dCopy full SHA for 1238d5d

File tree

-4

lines changed

-4

lines changed

Lines changed: 4 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ fi`
`22`	`22`	`LOG=dali.log`
`23`	`23`
`24`	`24`	`SECONDS=0`
`25`		`-torchrun --nproc_per_node=${NUM_GPUS} main.py -a resnet50 --b 256 --loss-scale 128.0 --workers 8 --lr=0.4 --fp16-mode --epochs2 ./2>&1\| tee$LOG`
	`25`	`+torchrun --nproc_per_node=${NUM_GPUS} main.py -a resnet50 --b 256 --loss-scale 128.0 --workers 8 --lr=0.4 --fp16-mode --epochs1 ./2>&1\| tee$LOG`
`26`	`26`
`27`	`27`	`RET=${PIPESTATUS[0]}`
`28`	`28`	`echo"Training ran in$SECONDS seconds"`
`@@ -31,9 +31,9 @@ if [[ $RET -ne 0 ]]; then`
`31`	`31`	`CLEAN_AND_EXIT 2`
`32`	`32`	`fi`
`33`	`33`
`34`		`-MIN_TOP1=20.0`
`35`		`-MIN_TOP5=40.0`
`36`		`-MIN_PERF=100`
	`34`	`+MIN_TOP1=10.0`
	`35`	`+MIN_TOP5=20.0`
	`36`	`+MIN_PERF=50`
`37`	`37`
`38`	`38`	`TOP1=$(grep"^##Top-1"$LOG\| awk'{print $2}')`
`39`	`39`	`TOP5=$(grep"^##Top-5"$LOG\| awk'{print $2}')`

Comments

(0)