未验证 提交 c9912f0a 编写于 作者: B Bin Lu 提交者: GitHub

Update vgg_variant.py

上级 34d9c859
......@@ -11,7 +11,7 @@ class SigmoidSuffix(paddle.nn.Layer):
self.origin_layer = origin_layer
self.sigmoid = Sigmoid()
def forward(self, *input, res_dict=None, **kwargs):
def forward(self, input, res_dict=None, **kwargs):
x = self.origin_layer(input)
x = self.sigmoid(x)
return x
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册