From ccb875df842567c717b45bd566c7ccec18c4ffd3 Mon Sep 17 00:00:00 2001 From: MatejRojec Date: Wed, 19 Jun 2024 19:38:33 +0200 Subject: [PATCH] fix hooks --- luxonis_train/nodes/efficientnet.py | 2 -- luxonis_train/nodes/mobilenetv2.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/luxonis_train/nodes/efficientnet.py b/luxonis_train/nodes/efficientnet.py index c284adc5..426df846 100644 --- 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 index 19daa033..db6cf879 100644 --- 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):