espnet3.components.data.collect_stats.CollectStatsRunner

source

EnvironmentProvider, , batch_size: int | None = None, async_mode: bool = False, async_specs_dir: str | Path = './_async_specs', async_result_dir: str | Path = './_async_results')

Bases: BaseRunner

Runner that executes collect-stats over batches of indices.

Example

>>> runner = CollectStatsRunner(provider)
>>> results = runner([[0, 1], [2, 3]])

Initialize BaseRunner object.

static forward(batch_indices: Iterable[int] | int, , dataset, model, collate_fn, device, write_collected_feats: bool = False, collect_stats_kwargs: Dict[str, Any] | None = None)

Process a batch of dataset indices and compute feature statistics.