We are running a survey to help us improve the experience for all of our members. If you see the survey appear, please take the time to tell us about your experience if you can.
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