Длина прогнозируемых значений не равна фактическим значениям. Керас ТПУ

Когда я пытаюсь предсказать, у меня длина либо меньше, либо больше фактического значения. Что я должен делать. Фактическая длина - 700.

Получил 704 следующим образом
y_val=model.predict(val_dataset,steps=(len(val_label)/BATCH_SIZE))

Получил 672 следующим образом
y_val=model.predict(val_dataset,steps=(len(val_label)//BATCH_SIZE))

Что мне сделать, чтобы оно было равно 700.
len(val_label) равно 700
BATCH_SIZE равно 32


python keras tensorflow tpu
person Talha Anwar    schedule 17.09.2020    source источник
comment
попробуйте BATCH_SIZE из 35   -  person Andrey    schedule 17.09.2020
comment
дюйм. Размер партии TPU рассчитывается таким образом. BATCH_SIZE = 4*strategy.num_replicas_in_sync Так что я не думаю, что смогу дожить до 35 лет   -  person Talha Anwar    schedule 17.09.2020
comment
просто попробовал простую модель с batch_size 35 на ТПУ - работает. Вы, наверное, имеете в виду рекомендуемый размер партии   -  person Andrey    schedule 17.09.2020
comment
Это может быть связано с tpu-pred-method-has-the-shape. Можете ли вы попробовать последнюю версию tensorflow 2.3.1 или tf-nightly   -  person Gagik    schedule 08.10.2020
comment
Решение @Andrey вызывает потерю NaN, я попробую решение Gagik, если оно сработает   -  person Talha Anwar    schedule 08.10.2020