[Fix] Fix PixelContrastCrossEntropyLoss (#1803)

PaddlePaddle · Mar 2, 2022 · 4ef3fb4 · 4ef3fb4
1 parent f665b18
commit 4ef3fb4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/paddleseg/models/losses/pixel_contrast_cross_entropy_loss.py b/paddleseg/models/losses/pixel_contrast_cross_entropy_loss.py
@@ -100,9 +100,12 @@ def _hard_anchor_sampling(self, X, y_hat, y):
                 elif num_hard >= n_view / 2:
                     num_easy_keep = num_easy
                     num_hard_keep = n_view - num_easy_keep
-                else:
+                elif num_easy >= n_view / 2:
                     num_hard_keep = num_hard
                     num_easy_keep = n_view - num_hard_keep
+                else:
+                    num_hard_keep = num_hard
+                    num_easy_keep = num_easy
 
                 indices = None
                 if num_hard > 0: