search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
1.64k stars 178 forks source link

resume training #338

Open hzy-del opened 1 week ago

hzy-del commented 1 week ago

在src/solver/det_solver.py中fit函数中验证改为了

if (epoch % 50)==0: test_stats, coco_evaluator = evaluate( module, self.criterion, self.postprocessor, self.val_dataloader, base_ds, self.device, self.output_dir )

            为什么训练没有问题，重新训练会报错：
            local variable 'test_stats' referenced before assignment