Skip to content

Commit

Permalink
support keras functional model and tuning fallback layers (#521)
Browse files Browse the repository at this point in the history
Signed-off-by: Clark Chin <xi2.chen@intel.com>
Co-authored-by: chensuyue <suyue.chen@intel.com>
Co-authored-by: Lv, Liang1 <liang1.lv@intel.com>
  • Loading branch information
3 people committed Mar 2, 2023
1 parent 4bd540c commit efd7379
Show file tree
Hide file tree
Showing 48 changed files with 25,228 additions and 1,372 deletions.
39 changes: 39 additions & 0 deletions examples/.config/model_params_keras.json
@@ -0,0 +1,39 @@
{
"keras": {
"resnetv2_50": {
"model_src_dir": "image_recognition/resnetv2_50/quantization/ptq",
"dataset_location": "/tf_dataset/dataset/imagenet",
"input_model": "/tf_dataset2/models/tensorflow/resnetv2_50_keras/saved_model",
"main_script": "main.py",
"batch_size": 32
},
"xception": {
"model_src_dir": "image_recognition/xception/quantization/ptq",
"dataset_location": "/tf_dataset/dataset/imagenet",
"input_model": "/tf_dataset2/models/tensorflow/xception_keras/saved_model/",
"main_script": "main.py",
"batch_size": 32
},
"mnist": {
"model_src_dir": "image_recognition/mnist/quantization/qat",
"dataset_location": "",
"input_model": "/tf_dataset2/models/tensorflow/mnist_keras/saved_model/",
"main_script": "main.py",
"batch_size": 32
},
"resnetv2_101": {
"model_src_dir": "image_recognition/resnetv2_101/quantization/ptq",
"dataset_location": "/tf_dataset/dataset/imagenet",
"input_model": "/tf_dataset2/models/tensorflow/resnetv2_101_keras/saved_model",
"main_script": "main.py",
"batch_size": 32
},
"mobilenet_v2": {
"model_src_dir": "image_recognition/mobilenet_v2/quantization/ptq",
"dataset_location": "/tf_dataset/dataset/imagenet",
"input_model": "/tf_dataset2/models/tensorflow/mobilenet_v2_keras/saved_model/",
"main_script": "main.py",
"batch_size": 32
}
}
}
8 changes: 4 additions & 4 deletions examples/README.md
Expand Up @@ -61,7 +61,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
<td>MobileNet V2</td>
<td>Image Recognition</td>
<td>Post-Training Static Quantization</td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/mobilenet_v2/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/mobilenet_v2/quantization/ptq">keras</a></td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/mobilenet_v2/quantization/ptq">pb</a> / <a href="./keras/image_recognition/mobilenet_v2/quantization/ptq">keras</a></td>
</tr>
<tr>
<td>MobileNet V3</td>
Expand Down Expand Up @@ -115,13 +115,13 @@ Intel® Neural Compressor validated examples with multiple compression technique
<td>ResNet V2 50</td>
<td>Image Recognition</td>
<td>Post-Training Static Quantization</td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_50/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/resnetv2_50/quantization/ptq">keras</a></td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_50/quantization/ptq">pb</a> / <a href="./keras/image_recognition/resnetv2_50/quantization/ptq">keras</a></td>
</tr>
<tr>
<td>ResNet V2 101</td>
<td>Image Recognition</td>
<td>Post-Training Static Quantization</td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_101/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/resnetv2_101/quantization/ptq">keras</a></td>
<td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_101/quantization/ptq">pb</a> / <a href="./keras/image_recognition/resnetv2_101/quantization/ptq">keras</a></td>
</tr>
<tr>
<td>ResNet V2 152</td>
Expand Down Expand Up @@ -193,7 +193,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
<td>Xception</td>
<td>Image Recognition</td>
<td>Post-Training Static Quantization</td>
<td><a href="./tensorflow/image_recognition/keras_models/xception/quantization/ptq">keras</a></td>
<td><a href="./keras/image_recognition/xception/quantization/ptq">keras</a></td>
</tr>
<tr>
<td>ResNet V2</td>
Expand Down

0 comments on commit efd7379

Please sign in to comment.