espnet.asr.pytorch_backend.asr_init.create_transducer_compatible_state_dict

source

espnet.asr.pytorch_backend.asr_init.create_transducer_compatible_state_dict(model_state_dict, encoder_type, encoder_units)

Create a compatible transducer model state dict for transfer learning.

If RNN encoder modules from a non-Transducer model are found in the pre-trained model state dict, the corresponding modules keys are renamed for compatibility.

• Parameters:
  • model_state_dict (Dict) – Pre-trained model state dict
  • encoder_type (str) – Type of pre-trained encoder.
  • encoder_units (int) – Number of encoder units in pre-trained model.
• Returns: Transducer compatible pre-trained model state dict.
• Return type: new_state_dict (Dict)