espnet2.torch_utils.load_pretrained_model.filter_state_dict
Less than 1 minute
espnet2.torch_utils.load_pretrained_model.filter_state_dict
espnet2.torch_utils.load_pretrained_model.filter_state_dict(dst_state: Dict[str, float | Tensor], src_state: Dict[str, float | Tensor])
Filter name, size mismatch instances between dicts.
- Parameters:
- dst_state – reference state dict for filtering
- src_state – target state dict for filtering