# Dataset Quantization

## Prerequisites
- python 3.7+
- pytorch 1.8
- torchcam
- prefetch-generator
- timm 0.6.5 (For Dataset Quantization)

## Instructions

`dataset_quantization` implements the quantization pipeline of the proposed method. 

`pytorch_cifar` is utilized for CIFAR10 evaluation. 

`pytorch_image_models` is utilized for ImageNet evaluation. 

Please refer to the documentations inside each direcotory for further running details. 

