Python 预测值的长度不等于实际值。凯拉斯TPU

当我尝试预测时，我得到的长度比实际值短或大。 我该怎么办。实际长度为700

我通过下面的方法得到了704

y\u val=model.predict（val\u数据集，步骤=（len（val\u标签）/批次大小））

我用下面的方法得到了672

y\u val=model.predict（val\u数据集，步骤=（len（val\u标签）//批量大小））

我该怎么做才能使它等于700。

len（val_标签）

is 700

---

批量大小为32

尝试批量大小为35英寸。TPU批量大小是这样计算的

BATCH_SIZE=4*strategy.num_replications_in_sync

所以我认为我做不到35，只是在TPU上尝试了一个批量大小为35的简单模型-它可以工作。你可能是说推荐的批量大小可能与此有关。你能试试最新版本的tensorflow 2.3.1或@Andrey solution产生NaN损耗吗？如果可行，我会试试Gagik solution