From cdecc0b437141b907ef146787a10fd3adee5db8c Mon Sep 17 00:00:00 2001
From: zhangyubo0722 <94225063+zhangyubo0722@users.noreply.github.com>
Date: Fri, 1 Sep 2023 23:36:58 +0800
Subject: [PATCH] fix hgnet kwargs (#2949)

---
 ppcls/arch/backbone/legendary_models/pp_hgnet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/backbone/legendary_models/pp_hgnet.py b/ppcls/arch/backbone/legendary_models/pp_hgnet.py
index 9ca386f0..24bf149c 100644
--- a/ppcls/arch/backbone/legendary_models/pp_hgnet.py
+++ b/ppcls/arch/backbone/legendary_models/pp_hgnet.py
@@ -207,7 +207,8 @@ class PPHGNet(TheseusLayer):
                  use_last_conv=True,
                  class_expand=2048,
                  dropout_prob=0.0,
-                 class_num=1000):
+                 class_num=1000,
+                 **kwargs):
         super().__init__()
         self.use_last_conv = use_last_conv
         self.class_expand = class_expand
-- 
GitLab