Skip to content

Commit

Permalink
[Fix] Import Error on StartkHead
Browse files Browse the repository at this point in the history
  • Loading branch information
nijkah committed Apr 17, 2023
1 parent 71ba688 commit 51add38
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mmtrack/models/track_heads/stark_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
import torch.nn.functional as F
from mmcv.cnn.bricks import ConvModule
from mmcv.cnn.bricks.transformer import build_positional_encoding
from mmdet.models.layers import Transformer
from mmengine.model import BaseModule
from mmengine.structures import InstanceData
from torch import Tensor, nn
Expand Down Expand Up @@ -177,7 +176,7 @@ def forward(self, x: Tensor) -> Tensor:


@MODELS.register_module()
class StarkTransformer(Transformer):
class StarkTransformer(BaseModule):
"""The transformer head used in STARK. `STARK.
<https://arxiv.org/abs/2103.17154>`_.
Expand Down

0 comments on commit 51add38

Please sign in to comment.