espnet2.beats.generate_beats_checkpoint.generate_beats_encoder_checkpoint

source

espnet2.beats.generate_beats_checkpoint.generate_beats_encoder_checkpoint(espnet_state_dict: dict, deepspeed_checkpoint: bool = False, lightning_checkpoint: bool = False, key_prefix: str = 'encoder.')

Extract encoder weights from ESPnet checkpoint.