问题:ValueError: Shape must be rank 1 but is rank 0 for ‘bn_conv1/Reshape_4’ (op: ‘Reshape’) with input shapes: [1,1,1,64], [].
情况:
1.keras是2.2.4版本(keras-gpu),后端为tensorflow
2.你在用channels_first且用了bn层(不是用别人模型而是自己搭的话,用channels_last可以回避这个问题)
网上看到的解决方案都是降级到GPU2.1.6 或CPU2.2.0 其实不用
解决方法
第一步: 找到tensorflow_backend.py文件
如果有很多tensorflow_backend.py文件,建议所有的都改一下
第二步:修改文件:
把这些行中的(-1)改成中括号[-1],图片是我改过的.这个是2.2.4版bn层源码错误…