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
2.61k stars 303 forks source link

Train in different epoch problem #241

Open chenhao324 opened 7 months ago

chenhao324 commented 7 months ago

Hello author, I used the pytorch version of the code for fine-tuning. Why is the result obtained by the first epoch significantly better than that obtained by the second epoch

lyuwenyu commented 7 months ago
  1. May be too little data, one epoch is enough.
  2. Try adjusting the learning rate
  3. ...
chenhao324 commented 7 months ago

My data is over a million, and I also trained using pre trained weight parameters. I will try to modify the learning rate. Thank you for your answer