ORT squeezenet_qdq support new api
},
"squeezenet_qdq": {
"model_src_dir": "image_recognition/onnx_model_zoo/squeezenet/quantization/ptq",
- "dataset_location": "/tf_dataset2/datasets/imagenet/ImagenetRaw/ILSVRC2012_img_val",
+ "dataset_location": "/tf_dataset2/datasets/imagenet/ImagenetRaw",
"input_model": "/tf_dataset2/models/onnx/squeezenet/squeezenet1.0-13.onnx",
- "yaml": "squeezenet_qdq.yaml",
- "strategy": "basic",
- "batch_size": 1,
- "new_benchmark": true
+ "main_script": "main.py",
+ "batch_size": 1
},
"caffenet_qdq": {
"model_src_dir": "image_recognition/onnx_model_zoo/caffenet/quantization/ptq",
