Skip to content

Commit e78d369

Browse files
committed
[Fix] Import Error on StarkHead
1 parent 71ba688 commit e78d369

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

mmtrack/models/track_heads/stark_head.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import torch.nn.functional as F
77
from mmcv.cnn.bricks import ConvModule
88
from mmcv.cnn.bricks.transformer import build_positional_encoding
9-
from mmdet.models.layers import Transformer
109
from mmengine.model import BaseModule
1110
from mmengine.structures import InstanceData
1211
from torch import Tensor, nn
@@ -177,7 +176,7 @@ def forward(self, x: Tensor) -> Tensor:
177176

178177

179178
@MODELS.register_module()
180-
class StarkTransformer(Transformer):
179+
class StarkTransformer(BaseModule):
181180
"""The transformer head used in STARK. `STARK.
182181
183182
<https://arxiv.org/abs/2103.17154>`_.

0 commit comments

Comments
 (0)