From 69bd10858d878024301f08846437c86f44401c17 Mon Sep 17 00:00:00 2001
From: Liyulingyue <852433440@qq.com>
Date: Mon, 14 Aug 2023 20:58:14 +0800
Subject: [PATCH] rm un-used num_class

---
 paddleseg/models/losses/focal_loss.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddleseg/models/losses/focal_loss.py b/paddleseg/models/losses/focal_loss.py
index 38a866d522..6f77c9d3bc 100755
--- 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