glenn-jocher commited on
Commit
0c93ec7
·
unverified ·
1 Parent(s): 94705a9

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

Browse files
Files changed (1) hide show
  1. .github/workflows/ci-testing.yml +4 -4
.github/workflows/ci-testing.yml CHANGED
@@ -68,10 +68,10 @@ jobs:
68
  di=cpu # device
69
 
70
  # Train
71
- python train.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
72
  # Val
73
- python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di
74
- python val.py --img 128 --batch 16 --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,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 128 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx # export
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