espnet3.components.data.collect_stats.collect_stats_batch

source

espnet3.components.data.collect_stats.collect_stats_batch(idxs: List[int], model=None, dataset=None, collate_fn=None, device: device | None = None, write_collected_feats: bool = False, collect_stats_kwargs: Dict[str, Any] | None = None)

Process a batch of dataset indices and compute feature statistics.