lucasoskorep/tensordex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding todo

Browse Source
This commit is contained in:
Lucas
2022-06-03 15:59:18 -04:00
parent 6add80bd27
commit 01703639ac
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
6_convert_keras_to_tflite.py
+1
View File
@@ -31,6 +31,7 @@ for file in glob("./models/keras/*.hdf5"):
tflite_model = converter.convert() tflite_model = converter.convert()
with open(tflite_file, 'wb') as f: with open(tflite_file, 'wb') as f:
f.write(tflite_model) f.write(tflite_model)
# TODO: Verify the model performance after converting to TFLITE
# interpreter = tf.lite.Interpreter(model_path=tflite_file) # interpreter = tf.lite.Interpreter(model_path=tflite_file)
# single_acc, single_ll = get_metrics(single_gen, keras_model) # single_acc, single_ll = get_metrics(single_gen, keras_model)
# tf_single_acc, tf_single_ll = get_metrics(single_gen, tflite_model) # tf_single_acc, tf_single_ll = get_metrics(single_gen, tflite_model)