# 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
```