当我尝试保存和加载包含 LSTM 层的模型时,加载公共失败并显示ValueError:找不到匹配的函数来调用从 SavedModel 加载的函数.
class RegNet(Model):
def __init__(self,
intermediate_dim=50,
state_dim=9,
name='RegNet',
**kwargs):
super(RegNet, self).__init__()
self.d1 = Dense(intermediate_dim, activation='relu')
self.d2 = Dense(state_dim, activation='relu')
self.h = LSTM(state_dim, activation='sigmoid', return_sequences=True)
self.o = Dense(state_dim, activation='softmax')
def call(self, x):
x = self.d1(x)
x = self.d2(x)
x = self.h(x)
y = self.o(x)
return y
regNet = RegNet()
...
# Export the model to a SavedModel
regNet.save(regNet_ckpt_dir, save_format='tf')
# Recreate the exact same model
tf.keras.models.load_model(regNet_ckpt_dir)
错误报告:
> ValueError: Could not find matching function to call loaded from the SavedModel. Got:
Positional arguments (2 total):
* Tensor("x:0", shape=(None, 1, 20), dtype=float32)
* Tensor("training:0", shape=(), dtype=bool)
Keyword arguments: {}
Expected these arguments to match one of the following 4 option(s):
Option 1:
Positional arguments (2 total):
* TensorSpec(shape=(None, 1, 20), dtype=tf.float32, name='input_1')
* False
Keyword arguments: {}
Option 2:
Positional arguments (2 total):
* TensorSpec(shape=(None, 1, 20), dtype=tf.float32, name='x')
* False
Keyword arguments: {}
Option 3:
Positional arguments (2 total):
* TensorSpec(shape=(None, 1, 20), dtype=tf.float32, name='x')
* True
Keyword arguments: {}
Option 4:
Positional arguments (2 total):
* TensorSpec(shape=(None, 1, 20), dtype=tf.float32, name='input_1')
* True
Keyword arguments: {}
当我注释 LSTM 层时,加载命令将会成功。哪里有问题?我们无法在 TensorFlow 2.0 中保存和加载带有 LSTM 层的模型?