bitorch.quantizations.quantization_scheduler.MixLinearScheduling

class bitorch.quantizations.quantization_scheduler.MixLinearScheduling(quantizations: List[Quantization] = [], steps: int = 0)[source]

Methods

quantize

interpolates linearly between the output of the specified quantizations.

Attributes

name

quantize(x: Tensor) → Tensor[source]

interpolates linearly between the output of the specified quantizations.

Parameters:

x (torch.Tensor) – input tensor

Returns:

quantized output tensor

Return type:

torch.Tensor