我使用的是一个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.
我想不出来
目前没有回答
相关问题 更多 >
编程相关推荐