Skip to content

dert_head.py bug with mmdet2trt version 0.6.0 #132

@KaneiGi

Description

@KaneiGi

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

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions