使用yolov8官方代码训练RTDETR的时候报错:NotImplementedError: WARNING ⚠️ ‘YOLO‘ model does not support ‘_new‘ mode

使用官方ultralytics的代码复现RTDETR的时候报错:NotImplementedError: WARNING ⚠️ 'YOLO' model does not support '_new' mode for 'None' task yet.

在这里插入图片描述

原本我只是简单的修改了yaml文件,就报错上述问题。后面参考官方的一个issues:use yolov8-rtdetr.yaml show error
将训练文件修改为:
import warnings
warnings.filterwarnings('ignore')
from ultralytics import RTDETR

if __name__ == '__main__':
    model = RTDETR('/home/root1/miniconda3/envs/itoshi16/lib/python3.8/site-packages/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml')
    model.train(data='/home/root1/data/miya/yolov8/data/data.yaml',
                cache=False,
                imgsz=1024,
                epochs=500,
                batch=4,
                close_mosaic=10,
                workers=8,
                device='0',  # 更改此处
                optimizer='SGD', # using SGD
                # resume='', # last.pt path
                amp=False, # close amp
                # fraction=0.2,
                project='runs/train',
                name='rt',
                )

此时便可以直接开始训练了:

在这里插入图片描述

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值