code: def add_args(cls, parser):
      SpeechToTextTask.add_args(parser)
      parser.add_argument(
          "--task-type",
          type=str,
          default="lang",
          help="task type for head selection, lang or domain",
      )
      parser.add_argument(
          "--kl-weight",
          type=float,
          default=0.0,
          help="the weight of KL loss",
      )
  
file_path: fairseq/examples/attention_head_selection/src/speech_to_text_head_selection.py
imports: import torch
  from fairseq.optim.amp_optimizer import AMPOptimizer
  from fairseq.tasks import register_task
  from fairseq.tasks.speech_to_text import SpeechToTextTask
  from .data.speech_to_text_dataset_with_domain import SpeechToTextDatasetCreatorWithDomain
  from .loss.attention_head_selection import HeadSelectionLoss
label_desc: Use image node_id 1 for calling the SpeechToTextHeadSelectionTask obj's underlying member method code with example usage: obj.add_args(cls, parser) without return types