Datasets
Models
Downloads: 1
Diff of
/train_yolo_v8.py
[000000]
..
[ab8373]
Switch to unified view
| a | b/train_yolo_v8.py | ||
|---|---|---|---|
| 1 | from ultralytics import YOLO |
||
| 2 | |||
| 3 | |||
| 4 | |||
| 5 | model = YOLO('yolov8x.pt') |
||
| 6 | model.to('cuda') |
||
| 7 | |||
| 8 | results = model.train( |
||
| 9 | data="yolo.yaml", |
||
| 10 | imgsz=224, |
||
| 11 | epochs=500, |
||
| 12 | batch=8, |
||
| 13 | name='train_valid_4', |
||
| 14 | patience=0, |
||
| 15 | ) |