Solved by a_d in post #2 
              
              
                To define the vgg model as one sequential layer you could simply do - 
vgg = models.vgg16(true/false)
vgg_features = nn.Sequential(*list(vgg.children())[:-1]) #removed the Last linear layer

And in the forward method - 
out = vgg_features(input)