From 51add380ea8e172aebe2302538b3dee78d2c4883 Mon Sep 17 00:00:00 2001
From: nijkah <nijkah@gmail.com>
Date: Mon, 17 Apr 2023 13:03:23 +0900
Subject: [PATCH] [Fix] Import Error on StartkHead

---
 mmtrack/models/track_heads/stark_head.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mmtrack/models/track_heads/stark_head.py b/mmtrack/models/track_heads/stark_head.py
index 3dbf1adce..ea53f3949 100644
--- a/mmtrack/models/track_heads/stark_head.py
+++ b/mmtrack/models/track_heads/stark_head.py
@@ -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
@@ -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>`_.