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 __init__(
      self,
      larc_config: Dict[str, Any] = None,
      lr: float = 0.1,
      momentum: float = 0.0,
      weight_decay: float = 0.0,
      nesterov: bool = False,
      use_larc: bool = False,
  ):
      super().__init__(
          larc_config=larc_config,
          lr=lr,
          momentum=momentum,
          weight_decay=weight_decay,
          nesterov=nesterov,
          use_larc=False,
      )
      self._use_fsdp_larc = use_larc
  
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 1 to create a new SGD_FSDP object from inherited base classes: SGD with example: obj = SGD_FSDP(larc_config, lr, momentum, weight_decay, nesterov, use_larc)