removed extra code

luxonis · Sep 28, 2024 · faf6e63 · faf6e63
1 parent 30ae9da
commit faf6e63
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/luxonis_train/nodes/necks/reppan_neck/blocks.py b/luxonis_train/nodes/necks/reppan_neck/blocks.py
@@ -229,12 +229,6 @@ def __init__(
     def encode_block(self) -> nn.Module:
         return self._encode_block
 
-    def forward(self, x0: Tensor, x1: Tensor) -> Tensor:
-        x = self.downsample(x0)
-        x = torch.cat([x, x1], dim=1)
-        x = self.encode_block(x)
-        return x
-
 
 class CSPDownBlock(PANDownBlockBase):
     def __init__(
@@ -279,9 +273,3 @@ def __init__(
     @property
     def encode_block(self) -> nn.Module:
         return self._encode_block
-
-    def forward(self, x0: Tensor, x1: Tensor) -> Tensor:
-        x = self.downsample(x0)
-        x = torch.cat([x, x1], dim=1)
-        x = self.encode_block(x)
-        return x