lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ResNet18在pytorch上的权重问题

mgh233 opened this issue · comments

首先是模型在resume过程中出错,具体原因是模型的字典没有对应上。
截屏2024-03-21 14 59 11
如上图所示,下载得到的模型的权重对应在module中而不是model中,并且model应该在第一层级上而不是在ema中。
以上问题我通过修改key值能够解决。
其次是将修改后的模型进行test,得到的结果为0,如下图所示。
截屏2024-03-21 15 04 08
测试从pytorch代码中训练出来的模型则具有正常结果,如下图所示。
截屏2024-03-21 15 10 53

  1. 提供的paddle转的参数只包含ema的结果,不完全对应torch训练时候的所有的state
  2. eval的时候使用的是solver ema的module, 所以参数应该加载到solver的ema里 不应该是模型本身