mmdetection-to-tensorrt icon indicating copy to clipboard operation
mmdetection-to-tensorrt copied to clipboard

dert_head.py bug with mmdet2trt version 0.6.0

Open KaneiGi opened this issue 1 year ago • 1 comments

Describe the bug On line 15 of mmdet2trt/models/dense_heads/detr_head.py:self.positional_encoding = build_wrapper(module.positional_encoding) Since mmdet version 3, the parameter positional_encoding is moved out from DETRHead's scope, so the module does not the attribution of positional_encoding.

environment: torch=2.2.0 tensorrt=8.6.1 mmdetection=3.3.0 cuda=11.7 mmdet2trt=0.6.0

KaneiGi avatar Oct 18 '24 07:10 KaneiGi

Thanks for the report. I will fix it soon.

grimoire avatar Oct 29 '24 02:10 grimoire