This commit is contained in:
Jirka 2020-07-16 11:49:28 +02:00
parent f0d1e099e2
commit 3e6868d318
1 changed files with 2 additions and 2 deletions

View File

@ -60,9 +60,9 @@ jobs:
# train
python train.py --weights weights/$name.pt --cfg models/$name.yaml --epochs 1 --img 320 --device $di --batch-size 2
# detect official
python detect.py --weights weights/$name.pt --device $di --batch-size 2
python detect.py --weights weights/$name.pt --device $di
# detect custom
python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2
python detect.py --weights runs/exp0/weights/last.pt --device $di
# test official
python test.py --weights weights/$name.pt --device $di --batch-size 2
# test custom