OLive/examples/vgg/vgg_config.json

31 строка
978 B
JSON

{
"input_model": { "type": "ONNXModel", "model_path": "models/vgg.onnx" },
"data_configs": [
{
"name": "raw_data",
"type": "RawDataContainer",
"load_dataset_config": {
"data_dir": "data",
"input_names": [ "data" ],
"input_shapes": [ [ 1, 3, 224, 224 ] ],
"input_dirs": [ "." ],
"input_suffix": ".raw",
"input_order_file": "input_order.txt"
}
}
],
"passes": {
"snpe_conversion": {
"type": "SNPEConversion",
"input_names": [ "data" ],
"input_shapes": [ [ 1, 3, 224, 224 ] ],
"output_names": [ "vgg0_dense2_fwd" ]
},
"snpe_quantization": { "type": "SNPEQuantization", "enable_htp": true, "data_config": "raw_data" }
},
"log_severity_level": 0,
"clean_cache": true,
"cache_dir": "cache",
"output_dir": "outputs"
}