Correct exception for dilated models

Author: qubvel

segmentation_models_pytorch/encoders/densenet.py

@@ -53,7 +53,7 @@ def __init__(self, out_channels, depth=5, **kwargs):
         self._in_channels = 3
         del self.classifier
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError("DenseNet encoders do not support dilated mode " "due to pooling operation for downsampling!")
 
     def get_stages(self):

segmentation_models_pytorch/encoders/inceptionresnetv2.py

@@ -50,7 +50,7 @@ def __init__(self, out_channels, depth=5, **kwargs):
         del self.avgpool_1a
         del self.last_linear
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError(
             "InceptionResnetV2 encoder does not support dilated mode " "due to pooling operation for downsampling!"
         )

segmentation_models_pytorch/encoders/timm_res2net.py

@@ -24,7 +24,7 @@ def get_stages(self):
             self.layer4,
         ]
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError("Res2Net encoders do not support dilated mode")
 
     def forward(self, x):

segmentation_models_pytorch/encoders/timm_resnest.py

@@ -24,7 +24,7 @@ def get_stages(self):
             self.layer4,
         ]
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError("ResNest encoders do not support dilated mode")
 
     def forward(self, x):

segmentation_models_pytorch/encoders/vgg.py

@@ -48,7 +48,7 @@ def __init__(self, out_channels, config, batch_norm=False, depth=5, **kwargs):
         self._in_channels = 3
         del self.classifier
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError("'VGG' models do not support dilated mode due to Max Pooling" " operations for downsampling!")
 
     def get_stages(self):

segmentation_models_pytorch/encoders/xception.py

@@ -21,7 +21,7 @@ def __init__(self, out_channels, *args, depth=5, **kwargs):
 
         del self.fc
 
-    def make_dilated(self, stage_list, dilation_list):
+    def make_dilated(self, *args, **kwargs):
         raise ValueError(
             "Xception encoder does not support dilated mode " "due to pooling operation for downsampling!"
         )