ExponentialMovingAverageObserver

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

A MinMaxObserver with momentum support for min/max updating.

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

SyncMinMaxObserver

next

SyncExponentialMovingAverageObserver