code: def __init__(self, backbone, epoch):
      super().__init__()
      self.save_hyperparameters()
      self.backbone = timm.create_model(
          backbone, num_classes=481 * 2 * 3
      )
      self.epoch = epoch
      # self.loss = nn.MSELoss(reduction='mean')
      self.loss = nn.L1Loss(reduction="mean")
      # self.hard_mining = False
      self.hard_mining = False
      self.num_face = 1103
      self.num_eye = 481 * 2
  
file_path: insightface/reconstruction/gaze/models.py
imports: import pytorch_lightning
  from pytorch_lightning.callbacks import ModelCheckpoint
  from pytorch_lightning.callbacks import LearningRateMonitor
  from pytorch_lightning.loggers import TensorBoardLogger
  import torch
  import torch.nn
  import timm
label_desc: Use image node_id 1 to create a new GazeModel object from inherited base classes: pl with example: obj = GazeModel(backbone, epoch)