code: def _reset_stats():
      return {
          "train": {
              "correct": 0.0,
              "count": 0.0,
          },
          "val": {
              "correct": 0.0,
              "count": 0.0,
          },
      }
  
imports: import math
  from typing import Optional, Tuple
  import torch
  import torch.nn.functional
  import torchaudio
  import torchaudio.models.wav2vec2.components
  from dataset import _get_lengths_librilightlimited, _get_lengths_librispeech, BucketizeBatchSampler, CollateFnHubert, CollateFnLibriLightLimited, DistributedBatchSampler, HuBERTDataSet
  from lightning.pytorch import LightningModule
  from loss import hubert_loss
  from torch import Tensor
  from torch.optim.optimizer import Optimizer
  from torch.utils.data import DataLoader
label_desc: Use image node_id 2 for calling a global function with example usage: _reset_stats() and returns: dict