espnet.asr.pytorch_backend.asr_init.transfer_verification

source

espnet.asr.pytorch_backend.asr_init.transfer_verification(model_state_dict, partial_state_dict, modules)

Verify tuples (key, shape) for input model modules match specified modules.

• Parameters:
  • model_state_dict (Dict) – Main model state dict.
  • partial_state_dict (Dict) – Pre-trained model state dict.
  • modules (List) – Specified module(s) to transfer.
• Returns: Whether transfer learning is allowed.
• Return type: (bool)