update

ZhangGongjie · Mar 17, 2022 · 41da2b0 · 41da2b0
1 parent 176bc41
commit 41da2b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/transformer_encoder.py b/models/transformer_encoder.py
@@ -11,7 +11,6 @@
 import torch.nn as nn
 from torch import Tensor
 
-from models.ops.modules import MSDeformAttn
 from models.misc import _get_clones, _get_activation_fn
 
 
@@ -124,6 +123,7 @@ def __init__(self, args, activation='relu'):
         self.n_points = 4
 
         # self attention
+        from models.ops.modules import MSDeformAttn
         self.self_attn = MSDeformAttn(self.d_model, self.n_feature_levels, self.nheads, self.n_points)
         self.dropout1 = nn.Dropout(self.dropout)
         self.norm1 = nn.LayerNorm(self.d_model)