code: def after_epoch(self, epoch_data: EpochData, **kwargs) -> None:
          
          self._save_stats(epoch_data)
          super().after_epoch(epoch_data=epoch_data, **kwargs)
docs: Compute the specified aggregations and save them to the given epoch data.
  
          :param epoch_data: epoch data to be processed