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
2.23k stars 246 forks source link

Missing hubconf.py file? #435

Open Niknu opened 3 weeks ago

Niknu commented 3 weeks ago

Hi,

Is it an plan to create a hubconf.py file to make it is easy to use the torch.hub.load and use the pre-trained models?

lyuwenyu commented 3 weeks ago

see https://github.com/lyuwenyu/RT-DETR/pull/436

lyuwenyu commented 2 weeks ago

I don't know the reason, but according to the source code of 2.1, there is this class. You can check it in your local machine.

https://github.com/pytorch/pytorch/blob/release/2.1/torch/optim/lr_scheduler.py#L26

from torch.optim.lr_scheduler import LRScheduler
Niknu commented 2 weeks ago

Sorry, I have confused my self with having python 3.6 + 3.8 on same machine. Therefore I had these problems with not having the same right libs installed.