code: def wavernn_data_gen_fn():
      waveform = torch.rand(
          4, 1, (N_TIME - KERNEL_SIZE + 1) * HOP_LENGTH
      )
      specgram = torch.rand(4, 1, N_FREQ, N_TIME)
      return dict(waveform=waveform, specgram=specgram)
  
file_path: ColossalAI/tests/kit/model_zoo/torchaudio/torchaudio.py
imports: from functools import partial
  import torch
  import torchaudio.models
  from ..registry import ModelAttribute, model_zoo
label_desc: Use image node_id 3 for calling a global function with example usage: wavernn_data_gen_fn() and returns: dict