Skip to content

Commit

Permalink
default optimize to False since it can cause crashes
Browse files Browse the repository at this point in the history
  • Loading branch information
agrimagsrl committed Aug 16, 2020
1 parent da83e62 commit 5798ed1
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = '[email protected]',
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',
Expand Down
2 changes: 1 addition & 1 deletion tinymlgen/tinymlgen.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 5798ed1

Please sign in to comment.