class_docstr: Lightning module which trains a deep set model until convergence.
class_docstr_tok: ["Lightning","module","which","trains","a","deep","set","model","until","convergence","."]
code: def configure_optimizers(self) -> optim.Optimizer:
      return optim.Adam(
          self.model.parameters(),
          lr=1e-2,
          weight_decay=self.weight_decay,
      )
  
file_path: gluonts/src/gluonts/nursery/tsbench/src/tsbench/surrogate/torch/deepset_lightning_module.py
imports: from typing import List, Tuple
  import pytorch_lightning
  import torch
  from pytorch_lightning.callbacks import Callback, EarlyStopping
  from torch import nn, optim
  from .losses import ListMLELoss
label_desc: Use image node_id 2 for calling the DeepSetLightningModule obj's underlying member method code with example usage: obj.configure_optimizers() and returns: optim