使用官方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',
                )

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