Fix inplace error using clone (#5)

xiuqhou · Jul 23, 2024 · 3a310d8 · 3a310d8
1 parent 399023a
commit 3a310d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/bricks/relation_transformer.py b/models/bricks/relation_transformer.py
@@ -527,4 +527,4 @@ def forward(self, src_boxes: Tensor, tgt_boxes: Tensor = None):
             pos_embed = self.pos_func(pos_embed).permute(0, 3, 1, 2)
         pos_embed = self.pos_proj(pos_embed)
 
-        return pos_embed
+        return pos_embed.clone()