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?