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

4
ultralytics/yolo/utils/configs/default.yaml
View File

@ -3,8 +3,7 @@
# Train settings -------------------------------------------------------------------------------------------------------
model: null # i.e. yolov5s.pt
cfg: null # i.e. yolov5s.yaml
model: null # i.e. yolov5s.pt, yolo.yaml
data: null # i.e. coco128.yaml
epochs: 300
batch_size: 16
@ -70,6 +69,7 @@ mosaic: 1.0 # image mosaic (probability)
mixup: 0.0 # image mixup (probability)
copy_paste: 0.0 # segment copy-paste (probability)
label_smoothing: 0.0
# anchors: 3
# Hydra configs --------------------------------------------------------------------------------------------------------
hydra: