转换RT-DETR 为pth文件失败

[iodncookie]

大佬，参考readme_rtdetr.txt文件，下载转换权重wget https://bj.bcebos.com/v1/paddledet/models/rtdetr_r18vd_dec3_6x_coco.pdparams
运行
python tools/convert_weights.py -f exps/rtdetr/rtdetr_r18vd_6x_coco.py -c rtdetr_r18vd_dec3_6x_coco.pdparams -oc rtdetr_r18vd_dec3_6x_coco.pth -nc 80 --device gpu 报错。

======================== copy weights ======================== transpose param: name='transformer.decoder.layers.0.self_attn.in_proj_weight' transpose param: name='transformer.decoder.layers.0.self_attn.out_proj.weight' transpose param: name='transformer.decoder.layers.0.cross_attn.sampling_offsets.weight' transpose param: name='transformer.decoder.layers.0.cross_attn.attention_weights.weight' transpose param: name='transformer.decoder.layers.0.cross_attn.value_proj.weight' transpose param: name='transformer.decoder.layers.0.cross_attn.output_proj.weight' transpose param: name='transformer.decoder.layers.0.linear1.weight' transpose param: name='transformer.decoder.layers.0.linear2.weight' transpose param: name='transformer.decoder.layers.1.self_attn.in_proj_weight' transpose param: name='transformer.decoder.layers.1.self_attn.out_proj.weight' transpose param: name='transformer.decoder.layers.1.cross_attn.sampling_offsets.weight' transpose param: name='transformer.decoder.layers.1.cross_attn.attention_weights.weight' transpose param: name='transformer.decoder.layers.1.cross_attn.value_proj.weight' transpose param: name='transformer.decoder.layers.1.cross_attn.output_proj.weight' transpose param: name='transformer.decoder.layers.1.linear1.weight' transpose param: name='transformer.decoder.layers.1.linear2.weight' transpose param: name='transformer.decoder.layers.2.self_attn.in_proj_weight' transpose param: name='transformer.decoder.layers.2.self_attn.out_proj.weight' transpose param: name='transformer.decoder.layers.2.cross_attn.sampling_offsets.weight' transpose param: name='transformer.decoder.layers.2.cross_attn.attention_weights.weight' transpose param: name='transformer.decoder.layers.2.cross_attn.value_proj.weight' transpose param: name='transformer.decoder.layers.2.cross_attn.output_proj.weight' transpose param: name='transformer.decoder.layers.2.linear1.weight' transpose param: name='transformer.decoder.layers.2.linear2.weight' Traceback (most recent call last): File "tools/convert_weights.py", line 599, in <module> main(exp, args) File "tools/convert_weights.py", line 571, in main copy(name2, w, model_std) File "tools/convert_weights.py", line 65, in copy assert mul1 == mul2 AssertionError