Add missing name attribute to deeplabv3 & deeplabv3plus (#1214)

sanamorii · web-flow · commit e76ed01d49ff · 2025-08-08T13:35:23.000+01:00
diff --git a/segmentation_models_pytorch/decoders/deeplabv3/model.py b/segmentation_models_pytorch/decoders/deeplabv3/model.py
@@ -119,6 +119,7 @@ def __init__(
             )
         else:
             self.classification_head = None
+        self.name = "deeplabv3-{}".format(encoder_name)
 
     def load_state_dict(self, state_dict, *args, **kwargs):
         # For backward compatibility, previously Decoder module was Sequential
@@ -238,3 +239,4 @@ def __init__(
             )
         else:
             self.classification_head = None
+        self.name = "deeplabv3plus-{}".format(encoder_name)

Original file line number	Diff line number	Diff line change
`@@ -119,6 +119,7 @@ def __init__(`
`119`	`119`	`)`
`120`	`120`	`else:`
`121`	`121`	`self.classification_head = None`
	`122`	`+ self.name = "deeplabv3-{}".format(encoder_name)`
`122`	`123`
`123`	`124`	`def load_state_dict(self, state_dict, args, *kwargs):`
`124`	`125`	`# For backward compatibility, previously Decoder module was Sequential`
`@@ -238,3 +239,4 @@ def __init__(`
`238`	`239`	`)`
`239`	`240`	`else:`
`240`	`241`	`self.classification_head = None`
	`242`	`+ self.name = "deeplabv3plus-{}".format(encoder_name)`