一、原因
出现此问题的原因是线程
二、解决方式
方式1:
直接加上main函数
例如:
from ultralytics import YOLO
if __name__ == '__main__':
model = YOLO("yolov8s-cls.pt")
model.train(data='dataset/', epochs=100, batch=4, imgsz=640)
方式2:
不写main函数,定义线程
例如:
from ultralytics import YOLO
model = YOLO("yolov8s-cls.pt")
model.train(data='dataset/', epochs=100, batch=4, imgsz=640, worker=0)