From 3bf77ca180ddaff0bba1caebfd5687489a53c565 Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Thu, 16 Jul 2020 11:16:12 +0200
Subject: [PATCH] lower batch-size

---
 .github/workflows/ci-testing.yml | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
index 660efe8..659cb49 100644
--- a/.github/workflows/ci-testing.yml
+++ b/.github/workflows/ci-testing.yml
@@ -47,21 +47,28 @@ jobs:
         pip list
       shell: bash
 
-    - name: Download data
-      run: |
-        python3 -c "from utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')" && mv ./coco128 ../
+    #- name: Download data
+    #  run: |
+    #    python3 -c "from utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')" && mv ./coco128 ../
 
     - name: Tests
       run: |
         for x in yolov5s yolov5m yolov5l yolov5x # models
         do
           di=cpu # inference devices
-          python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di  # train
-          python detect.py --weights $x.pt --device $di  # detect official
-          python detect.py --weights runs/exp0/weights/last.pt --device $di  # detect custom
-          python test.py --weights $x.pt --device $di # test official
-          python test.py --weights runs/exp0/weights/last.pt --device $di # test custom
-          python models/yolo.py --cfg $x.yaml # inspect
-          python models/export.py --weights $x.pt --img 640 --batch 1 # export
+          # train
+          python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di --batch-size 2
+          # detect official
+          python detect.py --weights $x.pt --device $di --batch-size 2
+          # detect custom
+          python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2
+          # test official
+          python test.py --weights $x.pt --device $di --batch-size 2
+          # test custom
+          python test.py --weights runs/exp0/weights/last.pt --device $di  --batch-size 2
+          # inspect
+          python models/yolo.py --cfg $x.yaml
+          # export
+          python models/export.py --weights $x.pt --img 640 --batch 1
         done
       shell: bash