code: def prepare(self, param_groups):
      self.optimizer = torch.optim.SGD(
          param_groups,
          lr=self._lr,
          nesterov=self._nesterov,
          momentum=self._momentum,
          weight_decay=self._weight_decay,
      )
      if self._use_larc:
          try:
              from apex.parallel.LARC import LARC
          except ImportError:
              raise RuntimeError("Apex needed for LARC")
          self.optimizer = LARC(
              optimizer=self.optimizer, **self._larc_config
          )
  
file_path: ClassyVision/classy_vision/optim/sgd.py
imports: from typing import Any, Dict
  import torch.optim
  from .None import ClassyOptimizer, register_optimizer
label_desc: Use image node_id 2 for calling the SGD obj's underlying member method code with example usage: obj.prepare(param_groups) without return types