Skip to content

Commit bab5cce

Browse files
authored
Update model.py
1 parent 4bd8d38 commit bab5cce

File tree

1 file changed

+1
-1
lines changed
  • segmentation_models_pytorch/unetplusplus

1 file changed

+1
-1
lines changed

segmentation_models_pytorch/unetplusplus/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,5 +86,5 @@ def __init__(
8686
else:
8787
self.classification_head = None
8888

89-
self.name = "u-{}".format(encoder_name)
89+
self.name = "unetplusplus-{}".format(encoder_name)
9090
self.initialize()

0 commit comments

Comments
 (0)