rm un-used num_class (PaddlePaddle#3451)

sherwincn · Aug 16, 2023 · 9c79cbe · 9c79cbe
1 parent 0dd7dc5
commit 9c79cbe
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/paddleseg/models/losses/focal_loss.py b/paddleseg/models/losses/focal_loss.py
@@ -96,9 +96,8 @@ class MultiClassFocalLoss(nn.Layer):
             and does not contribute to the input gradient. Default ``255``.
     """
 
-    def __init__(self, num_class, alpha=1.0, gamma=2.0, ignore_index=255):
+    def __init__(self, alpha=1.0, gamma=2.0, ignore_index=255):
         super().__init__()
-        self.num_class = num_class
         self.alpha = alpha
         self.gamma = gamma
         self.ignore_index = ignore_index