Spaces:
Runtime error
Runtime error
Update model/backbones.py
Browse files- model/backbones.py +0 -5
model/backbones.py
CHANGED
|
@@ -12,8 +12,3 @@ class MobileNetV2Backbone(nn.Module):
|
|
| 12 |
return self.model(x)
|
| 13 |
|
| 14 |
|
| 15 |
-
# Добавим SUPPORTED_BACKBONES прямо сюда (или перенеси в backbones/__init__.py, если нужно)
|
| 16 |
-
SUPPORTED_BACKBONES = {
|
| 17 |
-
'mobilenetv2': lambda _=None: MobileNetV2Backbone()
|
| 18 |
-
} # _=None позволяет игнорировать входной параметр
|
| 19 |
-
|
|
|
|
| 12 |
return self.model(x)
|
| 13 |
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|