amitpant7 / Python-script-to-evaluate-quantized-model-on-tests-set

Dequantizing the vitis ai quantized model to evalaute it's performance on the test set in pytorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Python-script-to-evaluate-quantized-model-on-tests-set

Dequantizing the Vitis ai quantized model to evaluate its performance on the test set in Pytorch.

  • Requires Vitis AI, I have used vitis-ai-pytorch docker container
  • Adjust the directory structure accordingly or change the dataset file path and model file path in the code.

example: evaluate model on subset of 1000 images from val

python eval_quant.py --subset_len 1000

About

Dequantizing the vitis ai quantized model to evalaute it's performance on the test set in pytorch.


Languages

Language:Python 100.0%