Add residual block between convolutional layer and FC layer

So my model looks like this:

class ResNet50(nn.Module):
    def __init__(self, **kwargs):
        resnet50 = torchvision.models.resnet50(pretrained=True)
        self.base= nn.Sequential(*list(resnet50.children())[:-2])
        setattr(self,"fc0", nn.Linear(2048,751)) 
        setattr(self,"fc1", nn.Linear(2048,1360))         

    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 = {}
        num_fcs = 2
        for i in range(num_fcs):
            clf_outputs["fc%d" %i] = getattr(self, "fc%d" %i)(f)

        return clf_outputs,f 

I want to add residual block after last convolution layer and FC1, how can i do that ?

1 Like

I have the same question, Do you solve it?