code: def __init__(
      self,
      use_adversarial: bool = True,
      lr: float = 0.0004,
      b1: float = 0.9,
      b2: float = 0.999,
      **kwargs,
  ):
      super().__init__()
      # self.save_hyperparameters()
  
      self.lr = lr
      self.b1 = b1
      self.b2 = b2
      self.use_adversarial = use_adversarial
      self.generator = Generator(input_channels=3, output_channels=3)
  
      self.discriminator = Discriminator()
      self.featuresnet = FeaturesNet()
  
file_path: BigDL/python/nano/client-application/VideoStylization/litmodels.py
imports: from typing import OrderedDict
  import torch
  import torch.nn
  import torch.nn.functional
  import pytorch_lightning
  from feature_model import FeaturesNet
  from original_models import Discriminator, Generator
label_desc: Use image node_id 1 to create a new LitModel object from inherited base classes: pl with example: obj = LitModel(use_adversarial, lr, b1, b2)