code: def conformer_data_gen_fn():
      lengths = torch.randint(1, 400, (4,))
      input = torch.rand(4, int(lengths.max()), INPUT_DIM)
      return dict(input=input, 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 1 for calling a global function with example usage: conformer_data_gen_fn() and returns: dict