Skip to content

Commit c0524e3

Browse files
authored
Fix complex loss name (#209)
1 parent 886a545 commit c0524e3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

segmentation_models/base/objects.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,9 @@ def __init__(self, loss, multiplier):
9191

9292
# resolve name
9393
if len(loss.__name__.split('+')) > 1:
94-
name = '{} * ({})'.format(multiplier, loss.__name__)
94+
name = '{}({})'.format(multiplier, loss.__name__)
9595
else:
96-
name = '{} * {}'.format(multiplier, loss.__name__)
96+
name = '{}{}'.format(multiplier, loss.__name__)
9797
super().__init__(name=name)
9898
self.loss = loss
9999
self.multiplier = multiplier
@@ -105,7 +105,7 @@ def __call__(self, gt, pr):
105105
class SumOfLosses(Loss):
106106

107107
def __init__(self, l1, l2):
108-
name = '{} + {}'.format(l1.__name__, l2.__name__)
108+
name = '{}_plus_{}'.format(l1.__name__, l2.__name__)
109109
super().__init__(name=name)
110110
self.l1 = l1
111111
self.l2 = l2

0 commit comments

Comments
 (0)