espnet2.main_funcs.collect_stats.collect_stats

source

espnet2.main_funcs.collect_stats.collect_stats(model: AbsESPnetModel | None, train_iter: Iterable[Tuple[List[str], Dict[str, Tensor]]], valid_iter: Iterable[Tuple[List[str], Dict[str, Tensor]]], output_dir: Path, ngpu: int | None, log_interval: int | None, write_collected_feats: bool) → None

Perform on collect_stats mode.

Running for deriving the shape information from data and gathering statistics. This method is used before executing train().