更换优化器 - Githubissues

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.31k stars 258 forks source link

Open todesti2 opened 4 months ago

todesti2 commented 4 months ago

作者您好！RT-DETR使用的是算法库内置的AdamW优化器我想尝试替换成其他的优化算法例如变色龙算法的

我多次尝试改optimizer.py、trainer.py 等文件，但是在输出学习率等各方面都遇到问题，请问您有什么建议吗？

又或者说，咱们的这个框架支持更改这些吗？

感激您的回复！

lyuwenyu commented 4 months ago

默认只支持下面的，但是通过register方式你可以注册任意的优化器

SGD = register(optim.SGD)
Adam = register(optim.Adam)
AdamW = register(optim.AdamW)

todesti2 commented 4 months ago

默认只支持下面的，但是通过register方式你可以注册任意的优化器
SGD = register(optim.SGD)
Adam = register(optim.Adam)
AdamW = register(optim.AdamW)

您的意思是本框架支持我自己写一个优化器并使用对么

lyuwenyu commented 4 months ago

是的

todesti2 commented 4 months ago

是的

作者您好！