code: def __init__(self, encoder, decoder):
      super().__init__()
      self.encoder = encoder
      self.decoder = decoder
  
file_path: aim/examples/pytorch_lightning_track.py
imports: import os
  import lightning.pytorch
  from aim.pytorch_lightning import AimLogger
  from torch import nn, optim, utils
  from torchvision.datasets import MNIST
  from torchvision.transforms import ToTensor
label_desc: Use image node_id 1 to create a new LitAutoEncoder object from inherited base classes: pl with example: obj = LitAutoEncoder(encoder, decoder)