espnet3.components.data.collect_stats.CollectStatsInferenceProvider

source

class espnet3.components.data.collect_stats.CollectStatsInferenceProvider(model_config, dataset_config, dataloader_config, mode: str, task: str | None = None, shard_idx: int | None = None, params: Dict[str, Any] | None = None)

Bases: EnvironmentProvider

EnvironmentProvider tailored for collect-stats jobs.

Initialize CollectStatsInferenceProvider object.

build_env_local() → Dict[str, Any]

Build the environment once on the driver for local inference.

build_worker_setup_fn()

Return a Dask worker setup function that builds dataset/model.