code: def __init__(self, global_stats_path: str, sp_model_path: str):
      self.sp_model = spm.SentencePieceProcessor(
          model_file=sp_model_path
      )
      self.train_data_pipeline = torch.nn.Sequential(
          FunctionalModule(_piecewise_linear_log),
          GlobalStatsNormalization(global_stats_path),
          FunctionalModule(partial(torch.transpose, dim0=1, dim1=2)),
          torchaudio.transforms.FrequencyMasking(27),
          torchaudio.transforms.FrequencyMasking(27),
          torchaudio.transforms.TimeMasking(100, p=0.2),
          torchaudio.transforms.TimeMasking(100, p=0.2),
          FunctionalModule(partial(torch.transpose, dim0=1, dim1=2)),
      )
  
imports: import json
  import math
  from functools import partial
  from typing import List
  import sentencepiece
  import torch
  import torchaudio
  from data_module import LibriSpeechDataModule
  from lightning import Batch
label_desc: Use image node_id 1 to create a new TrainTransform object  with example: obj = TrainTransform(global_stats_path, sp_model_path)