PyTorch Quantization Aware Training

Introduction

PyTorch quantization aware training example for ResNet.

Usages

Build Docker Image

$ docker build -f docker/pytorch.Dockerfile --no-cache --tag=pytorch:1.8.1 .

Run Docker Container

$ docker run -it --rm --gpus device=0 --ipc=host -v $(pwd):/mnt pytorch:1.8.1

Run ResNet

$ python cifar.py

References

• PyTorch Quantization Aware Training
• PyTorch Static Quantization
• PyTorch CIFAR10 Tutorial
• PyTorch Transfer Learning Tutorial
• PyTorch Quantization
• PyTorch Distributed Training