我的keras编码器-解码器代码在cpu上工作正常。但是在gpu上,我无法使用1以外的batch_size启动它!这很奇怪。即使我在这种拟合中不使用批处理大小参数,我也会得到:
tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1568]与[32,49]
(默认情况下,keras的代码为batch_size = 32)
我可以使用任何我想要的batch_size运行许多其他程序。哪个模块将一个二维对象与一维对象进行比较?
答案 0 :(得分:-1)
我的问题很相似,但是我得到了:
tensorflow.python.framework.errors_impl.InvalidArgumentError:
不兼容的形状:[64]与[128]。
我发现这可能是因为损失函数未执行降维操作。
因此,我尝试了tf.reduce_mean()
,它显示正常。