dbg

546d63d5 · weishengyu · c749bdf9 · 546d63d5
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppcls/arch/backbone/legendary_models/resnet.py ppcls/arch/backbone/legendary_models/resnet.py +1 -1

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/resnet.py
+++ b/ppcls/arch/backbone/legendary_models/resnet.py
@@ -312,7 +312,7 @@ class ResNet(TheseusLayer):

        self.avg_pool = AdaptiveAvgPool2D(1)
        self.flatten = nn.Flatten()
-
+        self.avg_pool_channels = self.num_channels[-1] * 2
        stdv = 1.0 / math.sqrt(self.avg_pool_channels * 1.0)
        self.fc = Linear(
            self.avg_pool_channels,