espnet.asr.pytorch_backend.asr_init.freeze_modules

source

espnet.asr.pytorch_backend.asr_init.freeze_modules(model, modules)

Freeze model parameters according to modules list.

• Parameters:
  • model (torch.nn.Module) – Main model.
  • modules (List) – Specified module(s) to freeze.
• Returns: Updated main model. model_params (filter): Filtered model parameters.
• Return type: model (torch.nn.Module)