class_docstr: Averaging function for EMA with configurable decay rate
  (Supplementary '1.11.7 Evaluator setup').
class_docstr_tok: ["Averaging","function","for","EMA","with","configurable","decay","rate","(","Supplementary","'","1.11.7","Evaluator","setup","'",")","."]
code: def __init__(self, decay_rate: float) -> None:
      self._decay_rate = decay_rate
  
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 1 to create a new swa_avg_fn object  with example: obj = swa_avg_fn(decay_rate)