Fix complex loss name (qubvel#209)

sundawei · Oct 4, 2019 · c0524e3 · c0524e3
1 parent 886a545
commit c0524e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/segmentation_models/base/objects.py b/segmentation_models/base/objects.py
@@ -91,9 +91,9 @@ def __init__(self, loss, multiplier):
 
         # resolve name
         if len(loss.__name__.split('+')) > 1:
-            name = '{} * ({})'.format(multiplier, loss.__name__)
+            name = '{}({})'.format(multiplier, loss.__name__)
         else:
-            name = '{} * {}'.format(multiplier, loss.__name__)
+            name = '{}{}'.format(multiplier, loss.__name__)
         super().__init__(name=name)
         self.loss = loss
         self.multiplier = multiplier
@@ -105,7 +105,7 @@ def __call__(self, gt, pr):
 class SumOfLosses(Loss):
 
     def __init__(self, l1, l2):
-        name = '{} + {}'.format(l1.__name__, l2.__name__)
+        name = '{}_plus_{}'.format(l1.__name__, l2.__name__)
         super().__init__(name=name)
         self.l1 = l1
         self.l2 = l2