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: Updates the moving average.
docstr_tok: ["Updates","the","moving","average","."]
code: def update_ema(self):
      """Updates the moving average."""
      self.ema()
  
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 3 for calling the ExponentialMovingAverageModule obj's underlying member method code with example usage: obj.update_ema() without return types