We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 71ba688 commit 51add38Copy full SHA for 51add38
mmtrack/models/track_heads/stark_head.py
@@ -6,7 +6,6 @@
6
import torch.nn.functional as F
7
from mmcv.cnn.bricks import ConvModule
8
from mmcv.cnn.bricks.transformer import build_positional_encoding
9
-from mmdet.models.layers import Transformer
10
from mmengine.model import BaseModule
11
from mmengine.structures import InstanceData
12
from torch import Tensor, nn
@@ -177,7 +176,7 @@ def forward(self, x: Tensor) -> Tensor:
177
176
178
179
@MODELS.register_module()
180
-class StarkTransformer(Transformer):
+class StarkTransformer(BaseModule):
181
"""The transformer head used in STARK. `STARK.
182
183
<https://arxiv.org/abs/2103.17154>`_.
0 commit comments