espnet.nets.pytorch_backend.transducer.utils.valid_aux_encoder_output_layers

source

espnet.nets.pytorch_backend.transducer.utils.valid_aux_encoder_output_layers(aux_layer_id: List[int], enc_num_layers: int, use_symm_kl_div_loss: bool, subsample: List[int]) → List[int]

Check whether provided auxiliary encoder layer IDs are valid.

Return the valid list sorted with duplicates removed.

• Parameters:
  • aux_layer_id – Auxiliary encoder layer IDs.
  • enc_num_layers – Number of encoder layers.
  • use_symm_kl_div_loss – Whether symmetric KL divergence loss is used.
  • subsample – Subsampling rate per layer.
• Returns: Valid list of auxiliary encoder layers.
• Return type: valid