bitorch.quantizations.dorefa.WeightDoReFa¶
- class bitorch.quantizations.dorefa.WeightDoReFa(bits: Optional[int] = None)[source]¶
Module for applying the dorefa function on weights.
Reference: “DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradients” Zouh et al. 2016, https://arxiv.org/abs/1606.06160
Methods
Initiates quantization bits.
DoReFas the tensor to desired bit resolution using weight dorefa.
Attributes
bit_width
name