standalone val (#56)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Ayush Chaurasia
2022-11-30 15:04:44 +05:30
committed by GitHub
parent 3a241e4cea
commit 5a52e7663a
16 changed files with 161 additions and 31 deletions

View File

@ -93,9 +93,12 @@ jobs:
echo "TODO"
- name: Test segmentation
shell: bash # for Windows compatibility
# TODO: redo val test without hardcoded weights
run: |
yolo task=segment mode=train model=yolov5n-seg.yaml data=coco128-seg.yaml epochs=1 img_size=64
yolo task=segment mode=val model=runs/exp/weights/last.pt data=coco128-seg.yaml img_size=64
- name: Test classification
shell: bash # for Windows compatibility
run: |
yolo task=classify mode=train model=resnet18 data=mnist160 epochs=1 img_size=32
yolo task=classify mode=val model=runs/exp2/weights/last.pt data=mnist160