Fixes

Signed-off-by: Janusz Lisiecki <[email protected]>
NVIDIA · Feb 24, 2025 · 40cf0e4 · 40cf0e4
1 parent 56086e5
commit 40cf0e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/qa/TL1_ssd_training/test.sh b/qa/TL1_ssd_training/test.sh
@@ -7,7 +7,7 @@ test_body() {
     NUM_GPUS=$(nvidia-smi -L | wc -l)
     export DATA_DIR=/data/coco/coco-2017/coco2017/
     export IS_TMP_DIR=0
-    if [ ! -f "/data/coco/coco-2017/coco2017/train2017/000000581929.jpg"] && [ -f "/data/coco/coco-2017/coco2017/train2017.zip"]; then
+    if [ ! -f "/data/coco/coco-2017/coco2017/train2017/000000581929.jpg" ] && [ -f "/data/coco/coco-2017/coco2017/train2017.zip" ]; then
         export DATA_DIR=$(mktemp -d)
         export IS_TMP_DIR=1
         cd ${DATA_DIR}
@@ -19,7 +19,7 @@ test_body() {
         unzip -q val2017.zip &
         unzip -q annotations_trainval2017.zip &
     fi
-    torchrun --nproc_per_node=${NUM_GPUS} main.py --backbone resnet50 --warmup 300 --bs 64 --eval-batch-size 8 --epochs 4 --data ${DATA_DIR}--data_pipeline dali --target 0.085
+    torchrun --nproc_per_node=${NUM_GPUS} main.py --backbone resnet50 --warmup 300 --bs 64 --eval-batch-size 8 --epochs 4 --data ${DATA_DIR} --data_pipeline dali --target 0.085
     ((IS_TMP_DIR)) && rm -rf ${DATA_DIR}
 }
 

diff --git a/qa/TL3_SSD_convergence/test_pytorch.sh b/qa/TL3_SSD_convergence/test_pytorch.sh
@@ -17,7 +17,7 @@ NUM_GPUS=$(nvidia-smi -L | wc -l)
 
 export DATA_DIR=/data/coco/coco-2017/coco2017/
 export IS_TMP_DIR=0
-if [ ! -f "/data/coco/coco-2017/coco2017/train2017/000000581929.jpg"] && [ -f "/data/coco/coco-2017/coco2017/train2017.zip"]; then
+if [ ! -f "/data/coco/coco-2017/coco2017/train2017/000000581929.jpg" ] && [ -f "/data/coco/coco-2017/coco2017/train2017.zip" ]; then
     export DATA_DIR=$(mktemp -d)
     export IS_TMP_DIR=1
     cd ${DATA_DIR}