From faf6e6377ae82a8dce9e35654878f0a73d3ff2ed Mon Sep 17 00:00:00 2001 From: klemen1999 Date: Sat, 28 Sep 2024 21:30:26 +0200 Subject: [PATCH] removed extra code --- luxonis_train/nodes/necks/reppan_neck/blocks.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/luxonis_train/nodes/necks/reppan_neck/blocks.py b/luxonis_train/nodes/necks/reppan_neck/blocks.py index 9f7eda2b..b7c05a0f 100644 --- 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