code: def configure_optimizers(self):
      return torch.optim.SGD(
          self.parameters(),
          lr=self.lr,
          momentum=self.momentum,
          weight_decay=self.weight_decay,
      )
  
file_path: mlflow/examples/pytorch/AxHyperOptimizationPTL/iris.py
imports: import pytorch_lightning
  import torch
  import torch.nn.functional
  from torch import nn
label_desc: Use image node_id 3 for calling the IrisClassification obj's underlying member method code with example usage: obj.configure_optimizers() and returns: torch