From c370d43bd24523c42c61303706861967122d7991 Mon Sep 17 00:00:00 2001 From: Ryan <71486185+Ryan-py@users.noreply.github.com> Date: Thu, 9 Nov 2023 23:07:30 +0300 Subject: [PATCH] Update tinymlgen.py "Optimization_update" --- tinymlgen/tinymlgen.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinymlgen/tinymlgen.py b/tinymlgen/tinymlgen.py index b018cde..003b556 100644 --- a/tinymlgen/tinymlgen.py +++ b/tinymlgen/tinymlgen.py @@ -7,7 +7,7 @@ def port(model, optimize=False, variable_name='model_data', pretty_print=False): converter = tf.lite.TFLiteConverter.from_keras_model(model) if optimize: if isinstance(optimize, bool): - optimizers = [tf.lite.Optimize.OPTIMIZE_FOR_SIZE] + optimizers = [tf.lite.Optimize.DEFAULT] else: optimizers = optimize converter.optimizations = optimizers @@ -36,4 +36,4 @@ def port(model, optimize=False, variable_name='model_data', pretty_print=False): #endif ''' - return preamble + c \ No newline at end of file + return preamble + c