Hi guys, I need to know the size of the “In Features” (before Fully connected Layer) of resNet 50.
Is it 7x7x2048? Is the right size?
And if I want this “Image” this code:
class ResNetMultiHead(nn.Moule):
    def __init__(self):
        super(ResNetMultiHead, self).__init__()

        self.model = torchvison.models.resnet50()
        self.model.fc = nn.identity()

    def forward(self, x):

        head1 = self.model(x)
        head2 = head1
        head3 = head1

        return head1, head2, head3

Is it right?
So each head is 7x7x2048?