I was using dealing with FC layers initially:
class ResNet50(nn.Module):
    def __init__(self, **kwargs):
        super(ResNet50,self).__init__()
  )
       
    def forward(self,x):
        x = self.base(x)
        clf_outputs = {}
        num_fcs = 2
        
       

Results weren’t satisfactory as discussed here 29 so I am dealing with conv layers as well:
I’m getting this error RuntimeError: input has less dimensions than expected . I’ve tried debugging it in many ways but couldn’t seem to find the fix. Earlier I wasn’t getting it. What’s going wrong ?