espnet2.train.reporter.SubReporter

source

class espnet2.train.reporter.SubReporter(key: str, epoch: int, total_count: int)

Bases: object

This class is used in Reporter.

See the docstring of Reporter for the usage.

finished() → None

get_epoch() → int

get_total_count() → int

Returns the number of iterations over all epochs.

log_message(start: int | None = None, end: int | None = None) → str

measure_iter_time(iterable, name: str)

measure_time(name: str)

next()

Close up this step and reset state for the next step

register(stats: Dict[str, float | int | complex | Tensor | ndarray | Dict[str, float | int | complex | Tensor | ndarray] | None], weight: float | int | complex | Tensor | ndarray | None = None) → None

tensorboard_add_scalar(summary_writer, start: int | None = None)

wandb_log(start: int | None = None)