Hy guys, how can I extract the features in a resnet50 before the general average pooling? I need the image of 7x7x2048. If I put the FC in an nn.Identity in forward I only obtain the features vector. I need the image before the final pooling. How can I make?