cppflow icon indicating copy to clipboard operation
cppflow copied to clipboard

Loading h5 models

Open Psyhich opened this issue 3 years ago • 1 comments

Hi, I'm trying to use your library to load model in h5 format, but it's maybe ill formed or just not possible. I've created model using such python code:

stacked = tf.keras.layers.StackedRNNCells(\
    [tf.keras.layers.LSTMCell(lstm_count) for _ in range(lstm_layers_count)])
    
model = tf.keras.Sequential([
    tf.keras.layers.RNN(stacked),
    tf.keras.layers.Dense(outpust_count)
])

And I cannot save it in other formats because I will not be able to use functions of this model. So my question is it possible to load such model using your library or I should look for something else?

Psyhich avatar Oct 25 '22 02:10 Psyhich

Save the model in the saved_model format. This can be done by doing model.save("model_name"). Tensorflow will save the model in the saved_model format if you don't put .h5 in "model_name".

yochananscharf avatar Oct 25 '22 08:10 yochananscharf