class_docstr: AlphaFold SWA (Stochastic Weight Averaging) module wrapper.
class_docstr_tok: ["AlphaFold","SWA","(","Stochastic","Weight","Averaging",")","module","wrapper","."]
code: def update(self, alphafold: nn.Module) -> None:
      if self.enabled:
          self.averaged_model.update_parameters(model=alphafold)
  
file_path: apex/apex/contrib/test/openfold_triton/test_fused_adam_swa.py
imports: from itertools import chain
  import random
  import unittest
  import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the AlphaFoldSWA obj's underlying member method code with example usage: obj.update(alphafold) without return types