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 resnet.py

Browse Source
This commit is contained in:
cuicheng01 2021-05-28 11:32:51 +08:00 committed by GitHub
parent 90321ce38d
commit e6792d34da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/arch/backbone/legendary_models/resnet.py
View File

@ -294,7 +294,7 @@ class ResNet(TheseusLayer):
self.avg_pool_channels = self.num_channels[-1] * 2
stdv = 1.0 / math.sqrt(self.avg_pool_channels * 1.0)
self.out = Linear(
self.fc = Linear(
self.avg_pool_channels,
self.class_num,
weight_attr=ParamAttr(
@ -306,7 +306,7 @@ class ResNet(TheseusLayer):
x = self.blocks(x)
x = self.avg_pool(x)
x = paddle.reshape(x, shape=[-1, self.avg_pool_channels])
x = self.out(x)
x = self.fc(x)
return x