Constant name for final feature layer for easier visualization

[ThomasNorr]

Hello,
Is there a reason why the models sometimes use nn.AdaptiveAvgPool2d and sometimes no layer (F.avg_pool2d)?
Always using nn.AdaptiveAvgPool2d with a constant layer name like avgpool or maybe final_features would help with visualizing the activations as the lucent library uses the name of the layer. https://github.com/MadryLab/DebuggableDeepNetworks can for example not visualize activations on cifar resnets out of the box because for cifar the resnet has no such layer.

Greetings
Thomas