From 01703639ac556ab096b66246a7558464f4e81dfa Mon Sep 17 00:00:00 2001 From: Lucas Date: Fri, 3 Jun 2022 15:59:18 -0400 Subject: [PATCH] adding todo --- 6_convert_keras_to_tflite.py | 1 + 1 file changed, 1 insertion(+) diff --git a/6_convert_keras_to_tflite.py b/6_convert_keras_to_tflite.py index 12d3639..4e93838 100644 --- a/6_convert_keras_to_tflite.py +++ b/6_convert_keras_to_tflite.py @@ -31,6 +31,7 @@ for file in glob("./models/keras/*.hdf5"): tflite_model = converter.convert() with open(tflite_file, 'wb') as f: f.write(tflite_model) + # TODO: Verify the model performance after converting to TFLITE # interpreter = tf.lite.Interpreter(model_path=tflite_file) # single_acc, single_ll = get_metrics(single_gen, keras_model) # tf_single_acc, tf_single_ll = get_metrics(single_gen, tflite_model)