espnet.nets.pytorch_backend.transducer.utils.custom_torch_load
Less than 1 minute
espnet.nets.pytorch_backend.transducer.utils.custom_torch_load
espnet.nets.pytorch_backend.transducer.utils.custom_torch_load(model_path: str, model: Module, training: bool = True)
Load Transducer model with training-only modules and parameters removed.
- Parameters:
- model_path – Model path.
- model – Transducer model.