I am trying to follow the guide from a scientific paper but can not work out how to convert the model from keras to VGG16
The paper is found at http://cs230.stanford.edu/projects_spring_2019/reports/18681590.pdf 4. The model is describe at the end of page 2 and start of 3.
I am taking the pretrainined VGG16 as the parameter and creating an output with 196 classes
class NetworkV1_4(nn.Module):#http://cs230.stanford.edu/projects_spring_2019/reports/18681590.pdf
    def __init__(self, base, num_classes):
        super().__init__()

        self.base = base

        self.base.classifier = nn.Sequential(
            nn.Linear(in_features=25088, out_features=25088,bias=True),
            nn.ReLU(inplace=True),
            nn.Linear(in_features=25088, out_features=512,bias=True),
            nn.ReLU(inplace=True),

            nn.Dropout(0.5),
            nn.Linear(512, num_classes),
        )

    def forward(self, x):
        fc = self.base(x)
        return fc