code: def reset_num_updates(self):
      del self.num_updates
      self.register_buffer(
          "num_updates", torch.tensor(0, dtype=torch.int)
      )
  
file_path: colossalai/examples/images/diffusion/ldm/modules/ema.py
imports: import torch
  from torch import nn
label_desc: Use image node_id 2 for calling the LitEma obj's underlying member method code with example usage: obj.reset_num_updates() without return types