Missing hubconf.py file?

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

Missing hubconf.py file? #435

Open Niknu opened 2 months ago

Niknu commented 2 months 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 2 months ago

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

lyuwenyu commented 2 months 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 months 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.