xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update model initialization design, supports custom data/num_classes (#44)

Browse Source 
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-15 20:06:29 +05:30
committed by GitHub
parent 1f3aad86c1
commit 832ea56eb4
8 changed files with 67 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/ci.yaml vendored
View File

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