zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Faster `--img 64` CI tests (#4979)

This commit is contained in:
Glenn Jocher 2021-09-28 12:41:28 -07:00 committed by GitHub
parent 94705a9528
commit 0c93ec762a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/ci-testing.yml vendored
View File

@ -68,10 +68,10 @@ jobs:
di=cpu # device di=cpu # device
# Train # 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 # Val
python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di python val.py --img 64 --batch 32 --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 runs/train/exp/weights/last.pt --device $di
# Detect # Detect
python detect.py --weights ${{ matrix.model }}.pt --device $di python detect.py --weights ${{ matrix.model }}.pt --device $di
python detect.py --weights runs/train/exp/weights/last.pt --device $di python detect.py --weights runs/train/exp/weights/last.pt --device $di
@ -79,7 +79,7 @@ jobs:
# Export # Export
python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model
python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow 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
python - <<EOF python - <<EOF
import torch import torch