mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Update vgg_variant.py

Browse Source
This commit is contained in:
Bin Lu 2021-08-24 10:53:13 +08:00 committed by GitHub
parent 34d9c8593e
commit c9912f0a04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/arch/backbone/variant_models/vgg_variant.py
View File

@ -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