提交 316479d3 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix: add *args, **kwargs to init_net()

the argument may be "infer_add_softmax" used in exporting.
上级 c032293a
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册