Add extra layer to vGG

1

How to add exra two layer to existing VGG?Please provide the code sample.

2

i didnt get you,give us an example

3

@sai_tharun In trained Vgg contains 1000 classes but if I want to restrict it to 10 classes how to do that?

5

vgg.classifier[6] = nn.Linear(in_features, 10)