Fix typo bianary_crossentropy -> binary_crossentropy (#196)

qubvel · Sep 3, 2019 · 886a545 · 886a545
1 parent 7a3d888
commit 886a545
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/segmentation_models/base/functional.py b/segmentation_models/base/functional.py
@@ -251,7 +251,7 @@ def categorical_crossentropy(gt, pr, class_weights=1., class_indexes=None, **kwa
     return - backend.mean(output)
 
 
-def bianary_crossentropy(gt, pr, **kwargs):
+def binary_crossentropy(gt, pr, **kwargs):
     backend = kwargs['backend']
     return backend.mean(backend.binary_crossentropy(gt, pr))
 

diff --git a/segmentation_models/losses.py b/segmentation_models/losses.py
@@ -127,7 +127,7 @@ def __init__(self):
         super().__init__(name='binary_crossentropy')
 
     def __call__(self, gt, pr):
-        return F.bianary_crossentropy(gt, pr, **self.submodules)
+        return F.binary_crossentropy(gt, pr, **self.submodules)
 
 
 class CategoricalCELoss(Loss):