class_docstr: Exponential Moving Average over a nn.ModuleDict.
  
  You can switch to the EMA weights temporarily.
code: def state_dict(self):
      return {"state": self.state, "count": self.count}
  
file_path: audiocraft/audiocraft/optim/ema.py
imports: from collections import defaultdict
  import typing
  import torch
  import torch.nn
label_desc: Use image node_id 4 for calling the ModuleDictEMA obj's underlying member method code with example usage: obj.state_dict() and returns: dict