fix hgnet kwargs (#2950)

2023-09-01 23:45:04 +08:00 · 2023-09-01 23:45:04 +08:00 · a119eb4191
parent 5168209b3b
commit a119eb4191
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