JirkaB commited on
Commit
da6c7ce
·
1 Parent(s): 9a880a4
Files changed (1) hide show
  1. .github/workflows/ci-testing.yml +3 -0
.github/workflows/ci-testing.yml CHANGED
@@ -58,6 +58,9 @@ jobs:
58
 
59
  - name: Tests workflow
60
  run: |
 
 
 
61
  di=cpu # inference devices
62
  # train
63
  python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 2
 
58
 
59
  - name: Tests workflow
60
  run: |
61
+ # to run *.py. files in subdirectories
62
+ export PYTHONPATH="$PWD"
63
+ # define device
64
  di=cpu # inference devices
65
  # train
66
  python train.py --weights weights/${{ matrix.yolo5-model }}.pt --cfg models/${{ matrix.yolo5-model }}.yaml --epochs 1 --img 320 --device $di --batch-size 2