From 2031a349a85f7b14ccc76bd2cd536d77dafe231b Mon Sep 17 00:00:00 2001 From: Ajay Raj Date: Thu, 16 Jul 2026 14:56:16 +0530 Subject: [PATCH] fix: update model save format to .keras for Keras 3 compatibility --- tensorflow/lite/micro/examples/hello_world/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/micro/examples/hello_world/train.py b/tensorflow/lite/micro/examples/hello_world/train.py index 59483acf04a..cdb9098b2a4 100644 --- a/tensorflow/lite/micro/examples/hello_world/train.py +++ b/tensorflow/lite/micro/examples/hello_world/train.py @@ -120,7 +120,7 @@ def train_model(epochs, x_values, y_values): verbose=2) if FLAGS.save_tf_model: - model.save(FLAGS.save_dir, save_format="tf") + model.save(os.path.join(FLAGS.save_dir, "model.keras")) logging.info("TF model saved to %s", FLAGS.save_dir) return model @@ -138,4 +138,4 @@ def main(_): if __name__ == "__main__": - app.run(main) \ No newline at end of file + app.run(main)