code: def update_parameters(self, model: torch.nn.Module) -> None:
      self._num_updates += 1
      if self._use_lit:
          decay = min(
              self._ema_decay,
              (1 + self._num_updates) / (10 + self._num_updates),
          )
  
          # TODO: torch/optim/swa_utils.pyi needs to be updated
          # pyre-ignore Undefined attribute [16]: Module `torch.optim.swa_utils` has no attribute `get_ema_multi_avg_fn`.
          self.multi_avg_fn = get_ema_multi_avg_fn(decay)
      super().update_parameters(model)
  
imports: from typing import Callable, List, Literal, Optional
  import torch
  from torch.optim.swa_utils import AveragedModel, get_ema_multi_avg_fn, get_swa_multi_avg_fn
label_desc: Use image node_id 2 for calling the AveragedModel obj's underlying member method code with example usage: obj.update_parameters(model) without return types