From 316479d380fe2fcf69e4473c5cbf3fdefc77ffd4 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 24 Oct 2022 12:52:58 +0000
Subject: [PATCH] fix: add *args, **kwargs to init_net()

the argument may be "infer_add_softmax" used in exporting.
---
 ppcls/arch/backbone/base/theseus_layer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py
index a9f4c60c..bf53467c 100644
--- a/ppcls/arch/backbone/base/theseus_layer.py
+++ b/ppcls/arch/backbone/base/theseus_layer.py
@@ -49,7 +49,9 @@ class TheseusLayer(nn.Layer):
                  return_patterns=None,
                  return_stages=None,
                  freeze_befor=None,
-                 stop_after=None):
+                 stop_after=None,
+                 *args,
+                 **kwargs):
         # init the output of net
         if return_patterns or return_stages:
             if return_patterns and return_stages:
-- 
GitLab