it answered my question. thanks a lot. But what the point of reducing the feature map to (1,1) as can be seen in the line
AdaptiveAvgPool2d(output_size=(1, 1))
Wouldn’t it be better to have a bigger output_size?