Не удается сохранить в формате SavedModel Tensorflow

Question

Я пытаюсь сохранить свою модель ИНС в формате SavedModel. Я использовал следующую команду:

model.save("my_model")

Она должна была предоставить мне папку, а именно «my_model», которая содержит все saved_model.pb, переменные и ресурсы, вместо этого она дает мне файл HDF, а именно my_model. Я использую keras v.2.3.1 и tensorflow v.2.2.0

Вот немного моего кода:

from keras import optimizers
from keras import backend
from keras.models import Sequential
from keras.layers import Dense
from keras.activations import relu,tanh,sigmoid

network_layout = []
for i in range(3):
    network_layout.append(8)

model = Sequential()

#Adding input layer and first hidden layer
model.add(Dense(network_layout[0],  
                name = "Input",
                input_dim=inputdim,
                kernel_initializer='he_normal',
                activation=activation))

#Adding the rest of hidden layer
for numneurons in network_layout[1:]:
    model.add(Dense(numneurons,
                    kernel_initializer = 'he_normal',
                    activation=activation))

#Adding the output layer
model.add(Dense(outputdim,
                name="Output",
                kernel_initializer="he_normal",
                activation="relu"))

#Compiling the model
model.compile(optimizer=opt,loss='mse',metrics=['mse','mae','mape'])
model.summary()

#Training the model
history = model.fit(x=Xtrain,y=ytrain,validation_data=(Xtest,ytest),batch_size=32,epochs=epochs)
model.save('my_model')

Я прочитал документацию API на веб-сайте tenorflow и Я сделал то, что было сказано, чтобы использовать model.save ("my_model") без какого-либо расширения файла, но я не могу понять это правильно.

Мы будем очень признательны за вашу помощь. Большое спасибо!

Answer 1

Если вы хотите использовать формат сохраненной модели тензорного потока, используйте:

tms_model = tf.saved_model.save(model,"export/1")

Это создаст папку экспорта и подпапку 1 внутри нее. Внутри 1 папки вы можете увидеть активы, переменные и файл .pb.

Надеюсь, это вам поможет.

Не забудьте изменить импорт следующим образом from tensorflow.keras import optimizers