Skip to content

Commit 886a545

Browse files
chAwaterqubvel
authored andcommitted
Fix typo bianary_crossentropy -> binary_crossentropy (#196)
1 parent 7a3d888 commit 886a545

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

segmentation_models/base/functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def categorical_crossentropy(gt, pr, class_weights=1., class_indexes=None, **kwa
251251
return - backend.mean(output)
252252

253253

254-
def bianary_crossentropy(gt, pr, **kwargs):
254+
def binary_crossentropy(gt, pr, **kwargs):
255255
backend = kwargs['backend']
256256
return backend.mean(backend.binary_crossentropy(gt, pr))
257257

segmentation_models/losses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def __init__(self):
127127
super().__init__(name='binary_crossentropy')
128128

129129
def __call__(self, gt, pr):
130-
return F.bianary_crossentropy(gt, pr, **self.submodules)
130+
return F.binary_crossentropy(gt, pr, **self.submodules)
131131

132132

133133
class CategoricalCELoss(Loss):

0 commit comments

Comments
 (0)