fix hgnet kwargs (#2950)

a119eb41 · zhangyubo0722 · GitHub · 5168209b · a119eb41
隐藏空白更改
内联并排

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