SyncExponentialMovingAverageObserver

class SyncExponentialMovingAverageObserver(momentum=0.9, mode=QuantMode.SYMMERTIC, eps=1e-05, dtype='qint8', **kwargs)

A distributed version of ExponentialMovingAverageObserver.

Parameters

• momentum (float) – momentum ratio for min/max updating.

• mode (QuantMode) – set quantization mode.

• eps (float) – a initial maximum value to avoid division by zero problem.

• dtype (Union[str, QuantDtypeMeta]) – a string indicating which dtype to collect scale and zero_point of.

previous

ExponentialMovingAverageObserver

next

HistogramObserver