lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

训练问题

zHJ991014 opened this issue · comments

作者你好,请问我训练自己的训练集出现这个问题是什么原因
Traceback (most recent call last):
File "C:\Users\zhanghaojie\Desktop\tarin\sourceofRT-DETR\RT-DETR-main\rtdetr_pytorch\tools\train.py", line 48, in
main(args)
File "C:\Users\zhanghaojie\Desktop\tarin\sourceofRT-DETR\RT-DETR-main\rtdetr_pytorch\tools\train.py", line 34, in main
solver.fit()
File "C:\Users\zhanghaojie\Desktop\tarin\sourceofRT-DETR\RT-DETR-main\rtdetr_pytorch\tools..\src\solver\det_solver.py", line 37, in fit
train_stats = train_one_epoch(
^^^^^^^^^^^^^^^^
File "C:\Users\zhanghaojie\Desktop\tarin\sourceofRT-DETR\RT-DETR-main\rtdetr_pytorch\tools..\src\solver\det_engine.py", line 35, in train_one_epoch
for samples, targets in metric_logger.log_every(data_loader, print_freq, header):
File "C:\Users\zhanghaojie\Desktop\tarin\sourceofRT-DETR\RT-DETR-main\rtdetr_pytorch\tools..\src\misc\logger.py", line 239, in log_every
header, total_time_str, total_time / len(iterable)))
~~~~~~~~~~~^~~~~~~~~~~~~~~
ZeroDivisionError: float division by zero

数据没加载对吧 可以打印一下dataloader的长度 看着像是是0