nn.AdaptiveAvgPool2d理解

如果是nn.AdaptiveAvgPool2d((1,1)), 可以从原来的[2, 2048, 10, 10] 转变为[2, 2048, 1, 1]
如果是nn.AdaptiveAvgPool2d(7), 则可以从[2, 2048, 10, 10] 转变为[2, 2048, 7, 7]

自适应的全局平均池化~只需要给出初始大小和输出大小,中间的步骤就由函数自己运算得到。

你可能感兴趣的:(深度学习项目)