YOLOv8在训练的时候遇到在result.png中train或者val的loss图像没有内容

最新推荐文章于 2024-06-16 17:26:07 发布

qq_52749859

最新推荐文章于 2024-06-16 17:26:07 发布

阅读量1.1k

点赞数 3

文章标签： YOLO python 机器学习

本文链接：https://blog.csdn.net/qq_52749859/article/details/136531521

版权

def train():

    # Load a model
    model = YOLO('yolov8n.yaml')  # build a new model from YAML
    model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
    model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

    # Train the model
    results = model.train(data='datasets/jyz/data.yaml', epochs=300, imgsz=640, device=0, batch=4, workers=0, amp=False)
if __name__ == "__main__":
    train()

试试使用py文件来运行训练代码试一下，并且将amp参数设置为false,我也是无意中试了一下就好了，本人cuda版本11.7，显卡为1650Ti。