code: def __init__(
      self, hidden_dim: int = 128, learning_rate: float = 0.0001
  ):
      super().__init__()
      self.save_hyperparameters()
  
      self.l1 = torch.nn.Linear(28 * 28, self.hparams.hidden_dim)
      self.l2 = torch.nn.Linear(self.hparams.hidden_dim, 10)
  
      self.val_outptus = []
      self.test_outputs = []
  
file_path: pytorch-lightning/examples/pytorch/ipu/mnist_sample.py
imports: import torch
  from lightning.pytorch import LightningModule, Trainer
  from lightning.pytorch.demos.mnist_datamodule import MNISTDataModule
  from torch.nn import functional
label_desc: Use image node_id 1 to create a new LitClassifier object from inherited base classes: LightningModule with example: obj = LitClassifier(hidden_dim, learning_rate)