Update model/backbones/mobilenetv2.py

model/backbones/mobilenetv2.py

@@ -6,7 +6,7 @@ class MobileNetV2Backbone(nn.Module):
         super(MobileNetV2Backbone, self).__init__()
         # Используем актуальный способ загрузки предобученной модели
         self.model = mobilenet_v2(weights=MobileNet_V2_Weights.DEFAULT).features
+        self.enc_channels = [24, 32, 96, 160, 320]  # Каналы с разных уровней feature extractor
 
     def forward(self, x):
         return self.model(x)
-