We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4bd8d38 commit bab5cceCopy full SHA for bab5cce
segmentation_models_pytorch/unetplusplus/model.py
@@ -86,5 +86,5 @@ def __init__(
86
else:
87
self.classification_head = None
88
89
- self.name = "u-{}".format(encoder_name)
+ self.name = "unetplusplus-{}".format(encoder_name)
90
self.initialize()
0 commit comments