espnet.asr.pytorch_backend.asr_init.load_trained_modules
Less than 1 minute
espnet.asr.pytorch_backend.asr_init.load_trained_modules
espnet.asr.pytorch_backend.asr_init.load_trained_modules(idim, odim, args, interface=<class 'espnet.nets.asr_interface.ASRInterface'>)
Load ASR/MT/TTS model with pre-trained weights for specified modules.
- Parameters:
- idim (int) – Input dimension.
- odim (int) – Output dimension.
- Namespace (args) – Model arguments.
- interface (ASRInterface |MTInterface |TTSInterface) – Model interface.
- Returns: Model with pre-initialized weights.
- Return type: main_model (torch.nn.Module)