args

.github/workflows/ci-testing.yml

@@ -60,9 +60,9 @@ jobs:
           # train
           python train.py --weights weights/$name.pt --cfg models/$name.yaml --epochs 1 --img 320 --device $di --batch-size 2
           # detect official
-          python detect.py --weights weights/$name.pt --device $di --batch-size 2
+          python detect.py --weights weights/$name.pt --device $di
           # detect custom
-          python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2
+          python detect.py --weights runs/exp0/weights/last.pt --device $di
           # test official
           python test.py --weights weights/$name.pt --device $di --batch-size 2
           # test custom