Cli support (#50)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia
2022-11-23 16:07:27 +05:30
committed by GitHub
parent 4291b9c31c
commit 512a225ce8
9 changed files with 70 additions and 17 deletions

View File

@ -94,8 +94,8 @@ jobs:
- name: Test segmentation
shell: bash # for Windows compatibility
run: |
python ultralytics/yolo/v8/segment/train.py model=yolov5n-seg.yaml data=coco128-seg.yaml epochs=1 img_size=64
yolo task=segment mode=train model=yolov5n-seg.yaml data=coco128-seg.yaml epochs=1 img_size=64
- name: Test classification
shell: bash # for Windows compatibility
run: |
python ultralytics/yolo/v8/classify/train.py model=resnet18 data=mnist160 epochs=1 img_size=32
yolo task=classify mode=train model=resnet18 data=mnist160 epochs=1 img_size=32