Can't load .tflite for "abalone" model

TensorFlow General Discussion
,
1

[Google Colab]

I’ve added some code to save “abalone” model as .tflite file and then load it. It creates .tflite file, but then I can’t use it:

ValueError: Cannot set tensor: Dimension mismatch. Got 7 but expected 1 for dimension 1 of input 0.

What’s wrong?

3

Hi burbilog,

Have you checked input shape with netron.app?

5

Have you checked input shape

+1

This fixes it:

normalize = preprocessing.Normalization(input_shape=[7])

Keras was keeping a general shape there, [None], and tflite seems to assumes that unknown shapes are 1 (useful for batch size).

1 Like
7

Thanks, it works now.