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

paths

This commit is contained in:
Jirka 2020-07-16 12:06:18 +02:00
parent 9a880a4554
commit da6c7cea2b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -58,6 +58,9 @@ jobs:
- name: Tests workflow - name: Tests workflow
run: | run: |
# to run *.py. files in subdirectories
export PYTHONPATH="$PWD"
# define device
di=cpu # inference devices di=cpu # inference devices
# train # train
python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 2 python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 2