code: def main():
      experiment.create(
          name="mnist_lit_lightening", disable_screen=True
      )
      model = LitMNIST()
      trainer = pl.Trainer(
          gpus=1,
          max_epochs=3,
          progress_bar_refresh_rate=20,
          logger=LabMLLightningLogger(),
      )
      with experiment.start():
          trainer.fit(model)
  
imports: import pytorch_lightning
  import torch
  from pytorch_lightning.metrics.functional import accuracy
  from torch import nn
  from torch.nn import functional
  from torch.utils.data import DataLoader, random_split
  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 for calling a global function with example usage: main() without return types