firstvgg.py

class newVGG(VGG):
    def __init__(self,
                 features: nn.Module,
                 **kwargs: Any) -> None:
        super().__init__(features,**kwargs)
        
    def forward(self, x: torch.Tensor) -> torch.Tensor:
        x = self.features(x)
        return x