espnet2.train.spk_trainer.SpkTrainer

source

class espnet2.train.spk_trainer.SpkTrainer

Bases: Trainer

Trainer designed for speaker recognition.

Training will be done as closed set classification. Validation will be open set EER calculation.

classmethod extract_embed(model: Module, iterator: Iterable[Dict[str, Tensor]], reporter: SubReporter, options: TrainerOptions, distributed_option: DistributedOption, output_dir: str, custom_bs: int, average: bool = False) → None

classmethod validate_one_epoch(model: Module, iterator: Iterable[Dict[str, Tensor]], reporter: SubReporter, options: TrainerOptions, distributed_option: DistributedOption) → None