热度

Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!

2

Hello author: Thanks for your excellent contribution, I've encountered a bug and don't know how to solve it .... When I use the official yolov8.pt to prune-post-train, the whole process...

"Keyword Error Alert"

3

命令:(rtdetr) root@autodl-container-d8bc11a952-991f7f93:~/RT-DETR-main/rtdetr_pytorch# python tools/train.py -c configs/rtdetr/rtdetr_r18vd_6x_coco.yml Not init distributed mode. Start training Load PResNet18 state_dict Initial lr: [1e-05, 1e-05, 0.0001, 0.0001] loading annotations into memory... Done (t=0.19s) creating index... index...