Thanks to visit codestin.com Credit goes to pepy.tech
Quantization-aware training in PyTorch
pip install brevitas