From 5798ed1857b1f9dc126f94537169d219db2d7cf0 Mon Sep 17 00:00:00 2001 From: Agrimagsrl Date: Sun, 16 Aug 2020 16:41:17 +0200 Subject: [PATCH] default `optimize` to False since it can cause crashes --- setup.py | 4 ++-- tinymlgen/tinymlgen.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 3894007..f33b5aa 100644 --- a/setup.py +++ b/setup.py @@ -2,13 +2,13 @@ setup( name = 'tinymlgen', packages = ['tinymlgen'], - version = '0.2', + version = '0.3', license='MIT', description = 'Generate C code for microcontrollers from Tensorflow models', author = 'Simone Salerno', author_email = 'eloquentarduino@gmail.com', url = 'https://github.com/eloquentarduino/tinymlgen', - download_url = 'https://github.com/eloquentarduino/tinymlgen/archive/v_02.tar.gz', + download_url = 'https://github.com/eloquentarduino/tinymlgen/archive/v_03.tar.gz', keywords = ['ML', 'microcontrollers', 'tensorflow', 'machine learning'], install_requires=[ 'tensorflow', diff --git a/tinymlgen/tinymlgen.py b/tinymlgen/tinymlgen.py index 6789a09..b018cde 100644 --- a/tinymlgen/tinymlgen.py +++ b/tinymlgen/tinymlgen.py @@ -3,7 +3,7 @@ import tensorflow as tf -def port(model, optimize=True, variable_name='model_data', pretty_print=False): +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):