espnet.asr.pytorch_backend.asr.DistributedDictSummary

source

class espnet.asr.pytorch_backend.asr.DistributedDictSummary(device=None)

Bases: object

Distributed version of DictSummary.

This implementation is based on an official implementation below. https://github.com/chainer/chainer/blob/v6.7.0/chainer/reporter.py

To gather stats information from all processes and calculate exact mean values, this class is running AllReduce operation in compute_mean().

Initialize Distributed DictSummary.

add(d)

Add values of the same names.

compute_mean()

Compute Mean stats.