在使用tf.saved_model保存模型之后,load加载模型使用时,发现如下错误:
仔细观察错误情况,发现是输入参数不匹配的原因。
这里需要注意一下,tf.saved_model保存模型时,函数的输入名称需要和加载时函数输入名称相同,
我发现模型函数是:
def call(self, inputs, **kwargs):
而加载后我是这样调用的:
fake_logits = discriminator(input=samples,training=False)
这时候应该发现了 input和inputs一个字母之差,改成一样的之后,便可以正常运行了