- prepare imagenet100 dataset
cd ..
mkdir datasets
// download imagenet100 dataset from kaggle (see below)
- train resnet18 with imagenet100 dataset
cd base_model
python train.py
// 'best_model.pth' will be generated in checkpoint directory.
- Post-training quantization (PTQ)
cd tmo_ptq
python ptq_onnx_export.py
python onnx2trt.py
// a file 'resnet18_int8_ptq_bf.engine' will be generated in engine directory.