From 53c37113c34fa551f5b57de36be4154ccc2aabce Mon Sep 17 00:00:00 2001 From: reuvenp Date: Tue, 5 Dec 2023 11:36:45 +0200 Subject: [PATCH] fix --- tutorials/notebooks/example_keras_imagenet.ipynb | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tutorials/notebooks/example_keras_imagenet.ipynb b/tutorials/notebooks/example_keras_imagenet.ipynb index e0dd1be7f..2ae6080ee 100644 --- a/tutorials/notebooks/example_keras_imagenet.ipynb +++ b/tutorials/notebooks/example_keras_imagenet.ipynb @@ -396,14 +396,13 @@ "source": [ "import tempfile\n", "_, tflite_file_path = tempfile.mkstemp('.tflite') # Path of exported model\n", - "mct.exporter.keras_export_model(model=quantized_model, \n", - " save_model_path=tflite_file_path,\n", - " serialization_format=mct.exporter.KerasExportSerializationFormat.TFLITE,\n", - " quantization_format=mct.exporter.QuantizationFormat.FAKELY_QUANT)\n", + "mct.exporter.keras_export_model(model=quantized_model, save_model_path=tflite_file_path,\n", + " target_platform_capabilities=target_platform_cap,\n", + " serialization_format=mct.exporter.KerasExportSerializationFormat.TFLITE)\n", "\n", "_, keras_file_path = tempfile.mkstemp('.h5') # Path of exported model\n", - "mct.exporter.keras_export_model(model=quantized_model, \n", - " save_model_path=keras_file_path,\n", + "mct.exporter.keras_export_model(model=quantized_model, save_model_path=keras_file_path,\n", + " target_platform_capabilities=target_platform_cap,\n", " serialization_format=mct.exporter.KerasExportSerializationFormat.KERAS_H5)" ], "metadata": {