code: def __init__(self):
      super(MNISTModel, self).__init__()
      self.l1 = torch.nn.Linear(28 * 28, 10)
  
file_path: labml/samples/lightning/mnist.py
imports: import pytorch_lightning
  import torch
  from torch.nn import functional
  from torch.utils.data import DataLoader
  from torchvision import transforms
  from torchvision.datasets import MNIST
  from labml import lab, experiment
  from labml.utils.lightning import LabMLLightningLogger
label_desc: Use image node_id 1 to create a new MNISTModel object from inherited base classes: pl with example: obj = MNISTModel()