你好，如果想用model.save()保存模型，怎么修改代码

[ai15231]

[bubbliiiing]

载入模型后再save呗。emm遇到了什么困难呢。

[ai15231]

您好，非常感谢您的回复。 1.我用 tf.keras.models.save_model()直接保存pb报错，报错信息ValueError: The innermost dimension of input_shape must be defined, but saw: (None, None)。模型里面的一些维度是None不能保存，然后我又改了模型一些层数的维度，但是还有一层是None，使用summary看到是tf_op_layer_strided_slice (Tens [()] ） 这里是空。目前还不知道怎么解决。 2.也用过其他的方法，比如用model.save()保存模型结构和参数成h5，然后在用代码去转换成pb,但是也报错，说AssertionError: dense_class_21/Identity is not in graph,这个节点在图里找不到，但是我用一些工具看保存的h5是有这个节点的，推测可能转换代码里会话的构建出了问题，但还是没有能够解决。 这个模型我需要的是全连接之后，输出的[32,2048]走向分类的特征张量，保存下来的模型的输出是这个节点的输出就更好了，我真的很需要这个pb模型的，就是希望能够保存成pb格式的模型文件，希望楼主能够给点指导意见，谢谢🙏