KD_Lib.Quantization.dynamic¶
KD_Lib.Quantization.dynamic.dynamic_quantization module¶
-
class
KD_Lib.Quantization.dynamic.dynamic_quantization.
Dynamic_Quantizer
(model, test_loader, qconfig_spec=None)[source]¶ Bases:
KD_Lib.Quantization.common.base_class.Quantizer
Implementation of Dynamic Quantization for PyTorch models.
Parameters: - model (torch.nn.Module) – Model that needs to be quantized
- qconfig_spec (Qconfig_spec) – Qconfig spec
- test_loader (torch.utils.data.DataLoader) – DataLoader used for testing