蠢办法:退回旧版本
使用python+keras+tensorflow来设计深度学习网络并做训练是非常寻常的事情,但是博主今天重新跑一个之前跑过的程序的时候突然不停的收到如标题所示的错误提醒。
已经确认我这台 新电脑 安装了正确的pycharm, python。
那么问题很明显就出在keras 和 tensorflow的版本上。
经过确认,把这两个框架的版本呢退回到我当时的版本之后,程序可以顺利运行了。
至于新版本到底是怎么更新的,还有待查明。
Keras和TF新版本解决方案
之前我提到新版本下面你的LABEL输入之后可能会莫名其妙多出一个额外的dimension。
比如在我这里原本是(3000,2)的Tensor就会变成(1,3000,2)。
去年我发现这是新版本的通病,不过解决方法也很简单。
只需要使用reshape把这个额外的dimension去掉即可。
具体方法我相信大家都会。