Browse Source

paths

5.0
Jirka 4 years ago
parent
commit
da6c7cea2b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      .github/workflows/ci-testing.yml

+ 3
- 0
.github/workflows/ci-testing.yml View File

@@ -58,6 +58,9 @@ jobs:

- name: Tests workflow
run: |
# to run *.py. files in subdirectories
export PYTHONPATH="$PWD"
# define device
di=cpu # inference devices
# 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

Loading…
Cancel
Save