search

Zhen-Dong / HAWQ

Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.
MIT License
410 stars 83 forks source link

cifar-10 example？？ #7

Closed chenjun2hao closed 3 years ago

chenjun2hao commented 3 years ago

is there have example on cifar-10? thanks!

Zhen-Dong commented 3 years ago

Hi, thanks for your interest. There are no cifar-10 experiments in HAWQ because generally cifar-10 results are very dependent on fine-tuning. However, all the dataset related codes are in quant_train.py and are easy to change. For example, https://github.com/Zhen-Dong/HAWQ/blob/main/quant_train.py#L391 And the training strategy may also need to be modified accordingly (such as lr, batch size, epochs, etc).

chenjun2hao commented 3 years ago

thanks for your reply！