Convert my Pytorch model to Pytorch Lightning

pip install pytorch-lightningimport pytorch_lightning as pl

# an nn class can be converted to a pl class by replacing nn with plclass NeuralNet(nn.Module): # --> class NeuralNet(pl.LightningModule):    def __init__(self, input_size, num_classes):        super(NeuralNet, self).__init__()        self.fc1 = nn.Linear(input_size, 50)        self.fc2 = nn.Linear(50, num_classes)# --> specific functions belong to nn class should not be changed!    def forward(self, x):        out = self.fc1(x)        out = torch.sigmoid(out)        out = self.fc2(out)        return out