code: def wav2vec_data_gen_fn():
      batch_size, num_frames = 4, 400
      waveforms = torch.randn(batch_size, num_frames)
      lengths = torch.randint(0, num_frames, (batch_size,))
      return dict(waveforms=waveforms, lengths=lengths)
  
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 5 for calling a global function with example usage: wav2vec_data_gen_fn() and returns: dict