This discussion has been locked.
You can no longer post new replies to this discussion. If you have a question you can start a new discussion

Save tensorflow model for ArmNN

Hello,

I want to load a Tensorflow model on my ARM-device. The model is rather simple:

Layer (type) Output Shape Param #
=================================================================
conv2d (Conv2D) (None, 1, 32, 8) 120
_________________________________________________________________
re_lu (ReLU) (None, 1, 32, 8) 0
_________________________________________________________________
conv2d_1 (Conv2D) (None, 1, 32, 4) 32
_________________________________________________________________
re_lu_1 (ReLU) (None, 1, 32, 4) 0
_________________________________________________________________
max_pooling2d (MaxPooling2D) (None, 1, 16, 4) 0
_________________________________________________________________
conv2d_2 (Conv2D) (None, 1, 8, 16) 320
_________________________________________________________________
re_lu_2 (ReLU) (None, 1, 8, 16) 0
_________________________________________________________________
conv2d_3 (Conv2D) (None, 1, 8, 8) 128
_________________________________________________________________
re_lu_3 (ReLU) (None, 1, 8, 8) 0
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 1, 4, 8) 0
_________________________________________________________________
conv2d_4 (Conv2D) (None, 1, 1, 8) 264
_________________________________________________________________
re_lu_4 (ReLU) (None, 1, 1, 8) 0
_________________________________________________________________
conv2d_5 (Conv2D) (None, 1, 1, 2) 18
_________________________________________________________________
flatten (Flatten) (None, 2) 0
=================================================================

I trained the model using Keras + Tesnorflow 1.13.1 and used the same Tensorflow version during the build process of ArmNN. The model is stored in a Keras model file (HDF5) and then converted into an binary Tensorflow model (.pb) using the save_model function.

Unfortunately, I am not able to the model and get the following error message:

'Unsupported operation Max in tensorflow::GraphDef  at function LoadNodeDef [/armnn/armnn-19.02/src/armnnTfParser/TfParser.cpp:3237]'

Since i dont use any uncommon Layers I am a little bit supprised. Loading the MNIST example model (simple_mnist_tf.pb) works fine.

Is their any option i have to pay attention when saving the model? Or is their any documentation about that topic?

Thanks in advance!

Gerald