code: def __init__(self, model=None, lr=1.0, gamma=0.7, batch_size=32):
      super().__init__()
      self.save_hyperparameters(ignore="model")
      self.model = model or Net()
      try:
          self.test_acc = Accuracy()
      except TypeError:
          self.test_acc = Accuracy("binary")
  
imports: import sys
  import torch
  import torchvision.transforms
  from torch.nn import functional
  import torch.nn
  from torchmetrics import Accuracy
  from torchvision.datasets.mnist import MNIST
  from pytorch_lightning import LightningModule
  from clearml import Task
label_desc: Use image node_id 1 to create a new ImageClassifier object from inherited base classes: LightningModule with example: obj = ImageClassifier(model, lr, gamma, batch_size)