class_docstr: A version of the SGD optimizer whose options (such as LARC) work with FSDP
class_docstr_tok: ["A","version","of","the","SGD","optimizer","whose","options","(","such","as","LARC",")","work","with","FSDP"]
code: def prepare(self, param_groups):
      super().prepare(param_groups)
      if self._use_fsdp_larc:
          self.optimizer = LARC_FSDP(
              optimizer=self.optimizer,
              distributed_norm=True,
              **self._larc_config,
          )
  
imports: from typing import Any, Dict
  import torch
  import torch.distributed
  from classy_vision.optim import register_optimizer, SGD
  from vissl.utils.fsdp_utils import get_global_group
label_desc: Use image node_id 2 for calling the SGD_FSDP obj's underlying member method code with example usage: obj.prepare(param_groups) without return types