bug fix
Browse files
model_utils/efficientnet_config.py
CHANGED
|
@@ -300,7 +300,7 @@ class EfficientNetPreTrained(PreTrainedModel):
|
|
| 300 |
def __init__(
|
| 301 |
self,
|
| 302 |
config
|
| 303 |
-
)
|
| 304 |
super().__init__(config)
|
| 305 |
self.model = EfficientNet( dropout=config.dropout,
|
| 306 |
num_channels=config.num_channels,
|
|
|
|
| 300 |
def __init__(
|
| 301 |
self,
|
| 302 |
config
|
| 303 |
+
):
|
| 304 |
super().__init__(config)
|
| 305 |
self.model = EfficientNet( dropout=config.dropout,
|
| 306 |
num_channels=config.num_channels,
|