【发布时间】：2018-10-28 10:47:00
【问题描述】：
我的 keras 编码器-解码器代码在 cpu 上运行良好。但在 gpu 上，我无法使用 1 以外的 batch_size 启动它！这很奇怪。即使我在此拟合中不使用批量大小参数，我也会得到：
tensorflow.python.framework.errors_impl.InvalidArgumentError：不兼容的形状：[1568] 与 [32,49]
（默认情况下，keras 采用 batch_size=32）
我可以使用我想要的任何 batch_size 运行许多其他程序。哪个模块将二维对象与一维对象进行比较？
【问题讨论】：
-
遇到类似错误的人也可以查看此链接 github.com/keras-team/keras/issues/11749 - 更改使用的指标有助于解决问题。
标签： tensorflow keras lstm