modify dice_loss

77c6085e · fosaken · 3b995e1d · 77c6085e
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

pdseg/loss.py pdseg/loss.py +2 -0

未找到文件。
--- a/pdseg/loss.py
+++ b/pdseg/loss.py
@@ -78,4 +78,6 @@ def dice_loss(logit, label, ignore_mask=None, num_classes=2):
    label = fluid.layers.cast(label, 'int64')
    ignore_mask = fluid.layers.reshape(ignore_mask, [-1, 1])
    loss = fluid.layers.dice_loss(logit, label)
+    label.stop_gradient = True
+    ignore_mask.stop_gradient = True
    return loss