code: def update_epoch_count(self):
      self.epoch_count += 1
      self.local_step_count = 0
  
imports: import time
  import torch
  from numpy import mean
  from deepspeed.utils.logging import log_dist
  from deepspeed import comm
label_desc: Use image node_id 2 for calling the ThroughputTimer obj's underlying member method code with example usage: obj.update_epoch_count() without return types