code: def main(hparams: Namespace):
      # ------------------------
      # 1 INIT LIGHTNING MODEL
      # ------------------------
      model = SegModel(**vars(hparams))
  
      # ------------------------
      # 2 INIT TRAINER
      # ------------------------
      trainer = Trainer()
  
      # ------------------------
      # 3 START TRAINING
      # ------------------------
      trainer.fit(model)
  
imports: import os
  import random
  from argparse import ArgumentParser, Namespace
  import numpy
  import torch
  import torch.nn.functional
  import torchvision.transforms
  from lightning.pytorch import LightningModule, Trainer, cli_lightning_logo
  from PIL import Image
  from torch import nn
  from torch.utils.data import DataLoader, Dataset
label_desc: Use image node_id 2 for calling a global function with example usage: main(hparams) without return types