Solved by iArunava in post #11 
              
              
                Before passing it to the classifiers Flatten it! 
class VGG(nn.Module):
  def __init__(self):
    . . .
  def forward(self, x):
    x = self.features(x)
    x = self.avgpool(x)
    x = x.reshape(x.size(0), -1)
    return self.classifier(x), self.classifier1(x)