fix hooks

luxonis · Jun 19, 2024 · ccb875d · ccb875d
1 parent aecb063
commit ccb875d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/luxonis_train/nodes/efficientnet.py b/luxonis_train/nodes/efficientnet.py
@@ -28,8 +28,6 @@ def __init__(self, download_weights: bool = False, **kwargs):
             pretrained=download_weights,
         )
         efficientnet_lite0_model.classifier = nn.Identity()
-
-
         self.out_indices = [1, 2, 4, 6]
         self.backbone = efficientnet_lite0_model
 

diff --git a/luxonis_train/nodes/mobilenetv2.py b/luxonis_train/nodes/mobilenetv2.py
@@ -30,12 +30,10 @@ def __init__(self, download_weights: bool = False, **kwargs):
             weights="DEFAULT" if download_weights else None
         )
         mobilenet_v2.classifier = nn.Identity()
-
         self.out_indices = [3, 6, 13, 18]
         self.channels = [24, 32, 96, 1280]
         self.backbone = mobilenet_v2
 
-
     def forward(self, x: Tensor) -> list[Tensor]:
         outs = []
         for i, module in enumerate(self.backbone.features):