code: def data_gen_for_sequence_classification():
      # sequence classification data gen
      # `labels` is the label for sequence classification, 0 or 1
      data = data_gen()
      data["labels"] = torch.tensor([1], dtype=torch.int64)
      return data
  
file_path: colossalai/tests/kit/model_zoo/transformers/bert.py
imports: import torch
  import transformers
  from ..registry import ModelAttribute, model_zoo
label_desc: Use image node_id 4 for calling a global function with example usage: data_gen_for_sequence_classification() and returns: data