提交 5b46305d authored 作者: Jirka's avatar Jirka

fix tests

上级 a79633a8
...@@ -10,7 +10,7 @@ jobs: ...@@ -10,7 +10,7 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: 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] python-version: [3.7, 3.8]
# Timeout: https://stackoverflow.com/a/59076067/4521646 # Timeout: https://stackoverflow.com/a/59076067/4521646
...@@ -55,8 +55,8 @@ jobs: ...@@ -55,8 +55,8 @@ jobs:
run: | run: |
for x in yolov5s yolov5m yolov5l yolov5x # models for x in yolov5s yolov5m yolov5l yolov5x # models
do do
python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device 0,1 # train
di=cpu # inference devices 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 $x.pt --device $di # detect official
python detect.py --weights runs/exp0/weights/last.pt --device $di # detect custom python detect.py --weights runs/exp0/weights/last.pt --device $di # detect custom
python test.py --weights $x.pt --device $di # test official python test.py --weights $x.pt --device $di # test official
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论