code: def get_batch_size(train_dataloader):
      if train_dataloader.batch_size is not None:
          return train_dataloader.batch_size
      elif train_dataloader.batch_sampler is not None:
          if (
              train_dataloader.batch_sampler.micro_batch_size
              is not None
          ):
              return train_dataloader.batch_sampler.micro_batch_size
          else:
              raise ValueError(
                  f"Could not find batch_size from batch_sampler: {train_dataloader.batch_sampler}"
              )
      else:
          raise ValueError(
              f"Could not find batch_size from train_dataloader: {train_dataloader}"
          )
  
imports: from enum import Enum
  from typing import Optional, Tuple
  import librosa
  import matplotlib.pylab
  import numpy
  import torch
  from einops import rearrange
  from numba import jit, prange
  from nemo.collections.tts.torch.tts_data_types import DATA_STR2DATA_CLASS, MAIN_DATA_TYPES, WithLens
  from nemo.utils import logging
  from nemo.utils.decorators import deprecated
label_desc: Use image node_id 1 for calling a global function with example usage: get_batch_size(train_dataloader) and returns: train_dataloader, train_dataloader