cittitt的博客

原创 Keras报错：InvalidArgumentError: Invalid input_h shape

最近做实验，用到lstm（tf-gpu的keras）。 之前已经用自己的数据训练过，一切正常，没有出现bug，然而，当我重新放入一批新的样本进行训练时，却报错： ，错误定位到了模型对验证集进行预测那行。 想到应该是shape出错了，但反反复复查了好几遍，训练集和验证集除了样本数量，其他完全一样。于是在实验室师兄的建议下，查看了模型各层的参数情况： 对比一下之前训练集的模型参数情况： 明明用的是同一个模型，为什么用我最新的训练集训练完之后模型各层的output shape反而和之前不一样