diff --git a/ppcls/arch/backbone/legendary_models/pp_hgnet.py b/ppcls/arch/backbone/legendary_models/pp_hgnet.py
index 9ca386f0694cdf50772703543d89c4fb9cd5444f..24bf149c314380d8bd1be69d3695964d9492543e 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