class_docstr: Creates a module that uses the moving average weights of another module.
class_docstr_tok: ["Creates","a","module","that","uses","the","moving","average","weights","of","another","module","."]
docstr: Creates ExponentialMovingAverageModule instance with given hyperparameters.
  
  Args:
      module: a module for which to compute the moving average.
      momentum: the decay factor for the moving average.
      debias: bool indicating whether to use initialization bias correction.
      eps: small adjustment to prevent division by zero.
docstr_tok: ["Creates","ExponentialMovingAverageModule","instance","with","given","hyperparameters",".","Args",":","module",":","a","module","for","which","to","compute","the","moving","average",".","momentum",":","the","decay","factor","for","the","moving","average",".","debias",":","bool","indicating","whether","to","use","initialization","bias","correction",".","eps",":","small","adjustment","to","prevent","division","by","zero","."]
code: def __init__(
      self,
      module: Module,
      momentum: float = 0.999,
      debias: bool = False,
      eps: float = 1e-6,
  ):
      """Creates ExponentialMovingAverageModule instance with given hyperparameters.
  
      Args:
          module: a module for which to compute the moving average.
          momentum: the decay factor for the moving average.
          debias: bool indicating whether to use initialization bias correction.
          eps: small adjustment to prevent division by zero.
      """
      self.__wrapped__ = module
      self.ema = ExponentialMovingAverage(
          module.vars(), momentum=momentum, debias=debias, eps=eps
      )
  
file_path: objax/objax/optimizer/ema.py
imports: from typing import Callable, Tuple, List
  import jax.numpy
  from objax.module import Module, ModuleList
  from objax.typing import JaxArray
  from objax.util import class_name
  from objax.variable import RandomState, TrainRef, StateVar, TrainVar, VarCollection
label_desc: Use image node_id 1 to create a new ExponentialMovingAverageModule object from inherited base classes: Module with example: obj = ExponentialMovingAverageModule(module, momentum, debias, eps)