code: def prepare(self, param_groups) -> None:
      self.optimizer = torch.optim._multi_tensor.AdamW(
          param_groups,
          lr=self._lr,
          betas=self._betas,
          eps=self._eps,
          weight_decay=self._weight_decay,
          amsgrad=self._amsgrad,
      )
  
file_path: ClassyVision/classy_vision/optim/adamw_mt.py
imports: from typing import Any, Dict, Tuple
  import torch.optim
  from .None import ClassyOptimizer, register_optimizer
label_desc: Use image node_id 2 for calling the AdamWMT obj's underlying member method code with example usage: obj.prepare(param_groups) without return types