espnet2.speechlm.model.speechlm.speechlm_job.SpeechLMJobTemplate

source

class espnet2.speechlm.model.speechlm.speechlm_job.SpeechLMJobTemplate(config: Dict[str, Any])

Bases: AbsJobTemplate

Job template for SpeechLM training tasks.

This class implements the specific model and data processing configurations for speech language modeling tasks.

Initialize the SpeechLM job template.

• Parameters:config – Dictionary containing job configuration parameters.

build_model() → Module

Build the SpeechLM model.

• Returns: A SpeechLM model instance.

build_preprocessor() → Callable

Build the data collation function for SpeechLM.

• Returns: A callable function for collating SpeechLM batch data.