From 0c93ec762aec91e4fd91762870f06e786abc9b10 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 28 Sep 2021 12:41:28 -0700
Subject: [PATCH] Faster `--img 64` CI tests (#4979)

---
 .github/workflows/ci-testing.yml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
index da695395fe69..c44e23995c3b 100644
--- a/.github/workflows/ci-testing.yml
+++ b/.github/workflows/ci-testing.yml
@@ -68,10 +68,10 @@ jobs:
           di=cpu  # device
 
           # Train
-          python train.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
+          python train.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
           # Val
-          python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di
-          python val.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
+          python val.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --device $di
+          python val.py --img 64 --batch 32 --weights runs/train/exp/weights/last.pt --device $di
           # Detect
           python detect.py --weights ${{ matrix.model }}.pt --device $di
           python detect.py --weights runs/train/exp/weights/last.pt --device $di
@@ -79,7 +79,7 @@ jobs:
           # Export
           python models/yolo.py --cfg ${{ matrix.model }}.yaml  # build PyTorch model
           python models/tf.py --weights ${{ matrix.model }}.pt  # build TensorFlow model
-          python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx  # export
+          python export.py --img 64 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx  # export
           # Python
           python - <<EOF
           import torch