Skip to content

Commit 5798ed1

Browse files
committed
default optimize to False since it can cause crashes
1 parent da83e62 commit 5798ed1

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,13 @@
22
setup(
33
name = 'tinymlgen',
44
packages = ['tinymlgen'],
5-
version = '0.2',
5+
version = '0.3',
66
license='MIT',
77
description = 'Generate C code for microcontrollers from Tensorflow models',
88
author = 'Simone Salerno',
99
author_email = '[email protected]',
1010
url = 'https://github.com/eloquentarduino/tinymlgen',
11-
download_url = 'https://github.com/eloquentarduino/tinymlgen/archive/v_02.tar.gz',
11+
download_url = 'https://github.com/eloquentarduino/tinymlgen/archive/v_03.tar.gz',
1212
keywords = ['ML', 'microcontrollers', 'tensorflow', 'machine learning'],
1313
install_requires=[
1414
'tensorflow',

tinymlgen/tinymlgen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import tensorflow as tf
44

55

6-
def port(model, optimize=True, variable_name='model_data', pretty_print=False):
6+
def port(model, optimize=False, variable_name='model_data', pretty_print=False):
77
converter = tf.lite.TFLiteConverter.from_keras_model(model)
88
if optimize:
99
if isinstance(optimize, bool):

0 commit comments

Comments
 (0)