Faster `--img 64` CI tests (#4979)
Browse files
.github/workflows/ci-testing.yml
CHANGED
@@ -68,10 +68,10 @@ jobs:
|
|
68 |
di=cpu # device
|
69 |
|
70 |
# Train
|
71 |
-
python train.py --img
|
72 |
# Val
|
73 |
-
python val.py --img
|
74 |
-
python val.py --img
|
75 |
# Detect
|
76 |
python detect.py --weights ${{ matrix.model }}.pt --device $di
|
77 |
python detect.py --weights runs/train/exp/weights/last.pt --device $di
|
@@ -79,7 +79,7 @@ jobs:
|
|
79 |
# Export
|
80 |
python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model
|
81 |
python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model
|
82 |
-
python export.py --img
|
83 |
# Python
|
84 |
python - <<EOF
|
85 |
import torch
|
|
|
68 |
di=cpu # device
|
69 |
|
70 |
# Train
|
71 |
+
python train.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
|
72 |
# Val
|
73 |
+
python val.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --device $di
|
74 |
+
python val.py --img 64 --batch 32 --weights runs/train/exp/weights/last.pt --device $di
|
75 |
# Detect
|
76 |
python detect.py --weights ${{ matrix.model }}.pt --device $di
|
77 |
python detect.py --weights runs/train/exp/weights/last.pt --device $di
|
|
|
79 |
# Export
|
80 |
python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model
|
81 |
python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model
|
82 |
+
python export.py --img 64 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx # export
|
83 |
# Python
|
84 |
python - <<EOF
|
85 |
import torch
|