提交 546d63d5 编写于 作者: W weishengyu

dbg

上级 c749bdf9
...@@ -312,7 +312,7 @@ class ResNet(TheseusLayer): ...@@ -312,7 +312,7 @@ class ResNet(TheseusLayer):
self.avg_pool = AdaptiveAvgPool2D(1) self.avg_pool = AdaptiveAvgPool2D(1)
self.flatten = nn.Flatten() self.flatten = nn.Flatten()
self.avg_pool_channels = self.num_channels[-1] * 2
stdv = 1.0 / math.sqrt(self.avg_pool_channels * 1.0) stdv = 1.0 / math.sqrt(self.avg_pool_channels * 1.0)
self.fc = Linear( self.fc = Linear(
self.avg_pool_channels, self.avg_pool_channels,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册