mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix typo (#1667)

pull/614/merge
Xiaolin Wang 2022-06-14 11:20:31 +08:00 committed by GitHub
parent 43b4efb122
commit df434bed69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmseg/models/losses/dice_loss.py
View File

@ -35,7 +35,7 @@ def dice_loss(pred,
@weighted_loss
def binary_dice_loss(pred, target, valid_mask, smooth=1, exponent=2, **kwards):
def binary_dice_loss(pred, target, valid_mask, smooth=1, exponent=2, **kwargs):
assert pred.shape[0] == target.shape[0]
pred = pred.reshape(pred.shape[0], -1)
target = target.reshape(target.shape[0], -1)
@ -79,7 +79,7 @@ class DiceLoss(nn.Module):
loss_weight=1.0,
ignore_index=255,
loss_name='loss_dice',
**kwards):
**kwargs):
super(DiceLoss, self).__init__()
self.smooth = smooth
self.exponent = exponent
@ -94,7 +94,7 @@ class DiceLoss(nn.Module):
target,
avg_factor=None,
reduction_override=None,
**kwards):
**kwargs):
assert reduction_override in (None, 'none', 'mean', 'sum')
reduction = (
reduction_override if reduction_override else self.reduction)