bitorch.quantizations.quantization_scheduler

Implementation of a quantization scheduler which replaces quantization functions inside a given model during training. This module also contains various scheduling procedure implementations which can be extended in future versions

Classes

MixLinearScheduling([quantizations, steps])
Quantization_Scheduler(model, steps, ...[, ...])
ScheduledQuantizer([quantizations, steps])	Base class for scheduled quantizers to inherit from.
StepScheduling([quantizations, steps])