|
|
@@ -10,7 +10,7 @@ jobs: |
|
|
|
strategy: |
|
|
|
fail-fast: false |
|
|
|
matrix: |
|
|
|
os: [ubuntu-18.04, windows-2019, macOS-10.15] |
|
|
|
os: [ubuntu-18.04, macOS-10.15] #, windows-2019 |
|
|
|
python-version: [3.7, 3.8] |
|
|
|
|
|
|
|
# Timeout: https://stackoverflow.com/a/59076067/4521646 |
|
|
@@ -55,8 +55,8 @@ jobs: |
|
|
|
run: | |
|
|
|
for x in yolov5s yolov5m yolov5l yolov5x # models |
|
|
|
do |
|
|
|
python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device 0,1 # train |
|
|
|
di=cpu # inference devices |
|
|
|
python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di # train |
|
|
|
python detect.py --weights $x.pt --device $di # detect official |
|
|
|
python detect.py --weights runs/exp0/weights/last.pt --device $di # detect custom |
|
|
|
python test.py --weights $x.pt --device $di # test official |