I’m using ResNet50 and wanted to add FC layers in parallel
I’m constructing my architecture like this:
class ResNet50(nn.Module):
def __init__(self, **kwargs):
super(ResNet50,self).__init__()
resnet50 = torchvision.models.resnet50(pretrained=True)
self.base= nn.Sequential(*list(resnet50.children())[:-2])
def forward(self,x):
x = self.base(x)
x = F.avg_pool2d(x,x.size()[2:])
f = x.view(x.size(0),-1)
clf_outputs = {}
return clf_outputs,f
I'm getting this:
mAP: 20.7%
What could be the possible reason for this ?