fix bootstrapped cross entropy bug (PaddlePaddle#806)

lijian8 · Jan 27, 2021 · a4f7645 · a4f7645
1 parent f850954
commit a4f7645
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddleseg/models/losses/bootstrapped_cross_entropy.py b/paddleseg/models/losses/bootstrapped_cross_entropy.py
@@ -38,8 +38,9 @@ def __init__(self, min_K, loss_th, weight=None, ignore_index=255):
         self.ignore_index = ignore_index
         self.K = min_K
         self.threshold = loss_th
+        if weight is not None:
+            weight = paddle.to_tensor(weight, dtype='float32')
         self.weight = weight
-        self.ignore_index = ignore_index
 
     def forward(self, logit, label):