Open
Description
I tried using python 3.6 + LCE 0.6.2 and python 3.7/3.8 + LCE 0.7.0 to run the following code, and the tflite file generated has unexpected sizes:
For python 3.6 + LCE 0.6.2:
XNOR tflite: 88.9 MB
BinaryDenseNet37 tflite: 25.6 MB
For python 3.7/3.8 + LCE 0.7.0:
XNOR tflite: 235.2 MB
BinaryDenseNet37 tflite: 5.4 MB (this looks normal)
Do you know what is causing this and what will be a solution? Thanks a lot!
import tensorflow as tf
import larq_zoo as lqz
import larq as lq
input_tensor = tf.keras.layers.Input(shape=(224, 224, 3))
# model = lqz.literature.BinaryDenseNet37Dilated(input_tensor=input_tensor, weights="imagenet")
model = lqz.literature.XNORNet(input_tensor=input_tensor, weights="imagenet")
lq.models.summary(model, print_fn=None, include_macs=True)
import os
path = os.path.join(os.getcwd(), './tflite_models')
if not os.path.exists(path):
os.makedirs(path)
with open(os.path.join(path,name+'.tflite'), 'wb') as flatbuffer_file:
flatbuffer_bytes = lce.convert_keras_model(model)
flatbuffer_file.write(flatbuffer_bytes)
Activity