Tensorflow无法保存展开的RNN模型

2024-04-24 14:04:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用的是一个GRU RNN模型,我希望展开该模型以用于部署:

import tensorflow as tf
from tensorflow import keras

model = keras.models.Sequential(name="KWS")
model.add(keras.Input(shape=(30,13)))
model.add(keras.layers.GRU(30, activation='linear', name='GRU_layer', unroll=True))
model.add(keras.layers.Dense(20, activation='relu', name='dense_1'))
model.add(keras.layers.Dense(1, activation='sigmoid', name='output'))
model.compile('rmsprop','mean_squared_error', metrics=['accuracy'])

# [training]

当我试图保存它时:

model.save('/temp/', save_format='tf')

它会产生以下错误:

ValueError: Unrolling requires a fixed number of timesteps.

我想不出来


Tags: name模型importaddmodelsavelayerstf