glenn-jocher commited on
Commit
d78a79b
·
unverified ·
1 Parent(s): 2efa01d

Update ci-testing.yml

Browse files
Files changed (1) hide show
  1. .github/workflows/ci-testing.yml +3 -3
.github/workflows/ci-testing.yml CHANGED
@@ -65,15 +65,15 @@ jobs:
65
  # define device
66
  di=cpu # inference devices
67
  # train
68
- python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 2
69
  # detect official
70
  python detect.py --weights weights/${{ matrix.yolo5-model }}.pt --device $di
71
  # detect custom
72
  python detect.py --weights runs/exp0/weights/last.pt --device $di
73
  # test official
74
- python test.py --weights weights/${{ matrix.yolo5-model }}.pt --device $di --batch-size 1
75
  # test custom
76
- python test.py --weights runs/exp0/weights/last.pt --device $di --batch-size 1
77
  # inspect
78
  python models/yolo.py --cfg models/${{ matrix.yolo5-model }}.yaml
79
  # export
 
65
  # define device
66
  di=cpu # inference devices
67
  # train
68
+ python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 8
69
  # detect official
70
  python detect.py --weights weights/${{ matrix.yolo5-model }}.pt --device $di
71
  # detect custom
72
  python detect.py --weights runs/exp0/weights/last.pt --device $di
73
  # test official
74
+ python test.py --weights weights/${{ matrix.yolo5-model }}.pt --device $di --batch 8
75
  # test custom
76
+ python test.py --weights runs/exp0/weights/last.pt --device $di --batch 8
77
  # inspect
78
  python models/yolo.py --cfg models/${{ matrix.yolo5-model }}.yaml
79
  # export