fix hgnet kwargs (#2949)

cdecc0b4 · zhangyubo0722 · GitHub · ed674366 · cdecc0b4
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppcls/arch/backbone/legendary_models/pp_hgnet.py ppcls/arch/backbone/legendary_models/pp_hgnet.py +2 -1

未找到文件。
--- 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