fix hgnet kwargs (#2949)

2025-06-03 21:55:06 +08:00 · 2023-09-01 23:36:58 +08:00 · 2023-09-01 23:36:58 +08:00 · cdecc0b437
commit cdecc0b437
parent ed67436647
1 changed files with 2 additions and 1 deletions
--- 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